multigeom.datasets.tinyimagenet

TinyImageNet, the diffusion-training and test images.

The images are transformed by `multigeom.datasets.transforms` (64 x 64, single-channel,
in [-1, 1]). Expected layout, under `data_dir()` (see `multigeom.config`):

    tiny-imagenet-200/train/<wnid>/images/<image>.JPEG
        100,000 images: training of the diffusion model
    tiny-imagenet-200/val/images/<image>.JPEG
        10,000 images; the first 200, in sorted order, are the test images

Classes

TinyImageNetDataset(*args, **kwargs)

Single-channel 64x64 view of TinyImageNet, used as the diffusion-training image distribution.