Module `imageapply.data`

Expand source code

from abc import ABC, abstractmethod, abstractproperty
import numpy as np
import torch
from typing import TypeVar

T = TypeVar("T", np.ndarray, torch.Tensor)

class data_object(ABC):

    def __init__(self, data):
        self.data = data

    @abstractproperty
    def shape(self):
        pass

    @abstractmethod
    def copy(self):
        pass

    @abstractmethod
    def split(self, n_splits, axis):
        pass

    @abstractmethod
    def concatenate(self, axis):
        pass

    @abstractmethod
    def __getitem__(self, key):
        pass

class numpy_data(data_object):

    def __init__(self, data):
        super().__init__(data)

    @property
    def shape(self):
        return self.data.shape

    def copy(self):
        return numpy_data(self.data.copy())

    def split(self, n_splits, axis):
        return numpy_data(np.split(self.data, n_splits, axis=axis))

    def concatenate(self, axis):
        return numpy_data(np.concatenate(self.data, axis=axis))

    def __getitem__(self, key):
        return numpy_data(self.data[key])

Classes

class data_object (data)

Helper class that provides a standard way to create an ABC using inheritance.

Expand source code

class data_object(ABC):

    def __init__(self, data):
        self.data = data

    @abstractproperty
    def shape(self):
        pass

    @abstractmethod
    def copy(self):
        pass

    @abstractmethod
    def split(self, n_splits, axis):
        pass

    @abstractmethod
    def concatenate(self, axis):
        pass

    @abstractmethod
    def __getitem__(self, key):
        pass

Ancestors

abc.ABC

Subclasses

numpy_data

Instance variables

var shape

Expand source code

@abstractproperty
def shape(self):
    pass

Methods

def concatenate(self, axis)

Expand source code

@abstractmethod
def concatenate(self, axis):
    pass

def copy(self)

Expand source code

@abstractmethod
def copy(self):
    pass

def split(self, n_splits, axis)

Expand source code

@abstractmethod
def split(self, n_splits, axis):
    pass

class numpy_data (data)

Helper class that provides a standard way to create an ABC using inheritance.

Expand source code

class numpy_data(data_object):

    def __init__(self, data):
        super().__init__(data)

    @property
    def shape(self):
        return self.data.shape

    def copy(self):
        return numpy_data(self.data.copy())

    def split(self, n_splits, axis):
        return numpy_data(np.split(self.data, n_splits, axis=axis))

    def concatenate(self, axis):
        return numpy_data(np.concatenate(self.data, axis=axis))

    def __getitem__(self, key):
        return numpy_data(self.data[key])

Ancestors

data_object
abc.ABC

Instance variables

var shape

Expand source code

@property
def shape(self):
    return self.data.shape

Methods

def concatenate(self, axis)

Expand source code

def concatenate(self, axis):
    return numpy_data(np.concatenate(self.data, axis=axis))

def copy(self)

Expand source code

def copy(self):
    return numpy_data(self.data.copy())

def split(self, n_splits, axis)

Expand source code

def split(self, n_splits, axis):
    return numpy_data(np.split(self.data, n_splits, axis=axis))