trw.datasets.tiny_imagenet

Module Contents

Classes

TinyImageNet

Tiny ImageNet data set available from http://cs231n.stanford.edu/tiny-imagenet-200.zip.

Functions

create_tiny_imagenet_dataset(batch_size: int, num_images_per_class: int = 500, transforms_train: List[trw.transforms.Transform] = None, transforms_valid: List[trw.transforms.Transform] = None, nb_workers: int = 4, root: Optional[str] = None) → Tuple[trw.basic_typing.Datasets, trw.basic_typing.DatasetsInfo]

class trw.datasets.tiny_imagenet.TinyImageNet(root, split='train', num_images_per_class=500)

Bases: torch.utils.data.Dataset

Tiny ImageNet data set available from http://cs231n.stanford.edu/tiny-imagenet-200.zip.

Notes

The test valid is discarded since we do not have the test labels

__len__(self)

__getitem__(self, index)

static read_image(path)

trw.datasets.tiny_imagenet.create_tiny_imagenet_dataset(batch_size: int, num_images_per_class: int = 500, transforms_train: List[trw.transforms.Transform] = None, transforms_valid: List[trw.transforms.Transform] = None, nb_workers: int = 4, root: Optional[str] = None) → Tuple[trw.basic_typing.Datasets, trw.basic_typing.DatasetsInfo]