Dataloading
Labelprop Index / Labelprop / Dataloading
Auto-generated documentation for labelprop.DataLoading module.
- Dataloading
- BatchLabelPropDataModule
- FullScan
- LabelPropDataModule
- PreTrainingDataModule
- UnsupervisedScan
- to_one_hot
BatchLabelPropDataModule
Show source in DataLoading.py:299
Equivalent to LabelPropDataModule, but for multiple images
Signature
class BatchLabelPropDataModule(pl.LightningDataModule):
def __init__(
self, img_path_list, mask_path_list, lab="all", shape=(288, 288), z_axis=0
): ...
BatchLabelPropDataModule().setup
Show source in DataLoading.py:314
Signature
def setup(self, stage=None): ...
BatchLabelPropDataModule().train_dataloader
Show source in DataLoading.py:332
Signature
def train_dataloader(self, batch_size=1): ...
FullScan
Show source in DataLoading.py:25
Signature
class FullScan(data.Dataset):
def __init__(
self,
X,
Y,
lab="all",
shape=256,
selected_slices=None,
z_axis=-1,
hints=None,
isotropic=True,
): ...
FullScan().norm
Show source in DataLoading.py:137
Signature
def norm(self, x, z_axis): ...
FullScan().resample
Show source in DataLoading.py:127
Signature
def resample(self, X, Y, size): ...
LabelPropDataModule
Show source in DataLoading.py:195
Signature
class LabelPropDataModule(pl.LightningDataModule):
def __init__(
self,
img_path,
mask_path,
lab="all",
shape=(288, 288),
selected_slices=None,
z_axis=0,
hints=None,
): ...
LabelPropDataModule().setup
Show source in DataLoading.py:228
Signature
def setup(self, stage=None): ...
LabelPropDataModule().test_dataloader
Show source in DataLoading.py:259
Signature
def test_dataloader(self): ...
LabelPropDataModule().train_dataloader
Show source in DataLoading.py:256
Signature
def train_dataloader(self, batch_size=1): ...
PreTrainingDataModule
Show source in DataLoading.py:263
Signature
class PreTrainingDataModule(pl.LightningDataModule):
def __init__(self, img_list, shape=(288, 288), z_axis=0): ...
PreTrainingDataModule().setup
Show source in DataLoading.py:278
Signature
def setup(self, stage=None): ...
PreTrainingDataModule().train_dataloader
Show source in DataLoading.py:293
Signature
def train_dataloader(self, batch_size=None): ...
UnsupervisedScan
Show source in DataLoading.py:148
Signature
class UnsupervisedScan(data.Dataset):
def __init__(self, X, shape=256, z_axis=-1, name=""): ...
UnsupervisedScan().norm
Show source in DataLoading.py:184
Signature
def norm(self, x): ...
to_one_hot
Show source in DataLoading.py:12
One hot encoding of a label tensor Input: Y: label tensor dim: dimension where to apply one hot encoding
Signature
def to_one_hot(Y, n_labels, dim=1): ...