nntoolbox.vision.utils.data module

class nntoolbox.vision.utils.data.PairedDataset(data_1: torch.utils.data.dataset.Dataset, data_2: torch.utils.data.dataset.Dataset)

Bases: Generic[torch.utils.data.dataset.T_co]

Pair up two datasets, and allow users to sample a pair, one from each dataset

class nntoolbox.vision.utils.data.UnlabelledImageDataset(paths: Union[List[str], str], transform=None, img_dim=None)

Bases: Generic[torch.utils.data.dataset.T_co]

class nntoolbox.vision.utils.data.UnlabelledImageListDataset(paths: Union[List[str], str], transform=None, img_dim=None)

Bases: Generic[torch.utils.data.dataset.T_co]

Abstraction for a list of path to images without labels

class nntoolbox.vision.utils.data.UnsupervisedFromSupervisedDataset(dataset: torch.utils.data.dataset.Dataset, transform=None)

Bases: Generic[torch.utils.data.dataset.T_co]

Convert a supervisded dataset to an unsupervised dataset