did.utils.imagenet

Module Contents

Classes

Imagenette

Functions

load_imagenet(traindir, valdir, resize_size, batch_size, num_workers, n_classes=None, shuffle=False)

Attributes

IMAGENET_NORMALIZE

PATH_TO_IMAGENET_TRAIN

PATH_TO_IMAGENET_TEST

did.utils.imagenet.IMAGENET_NORMALIZE
did.utils.imagenet.PATH_TO_IMAGENET_TRAIN =
did.utils.imagenet.PATH_TO_IMAGENET_TEST =
class did.utils.imagenet.Imagenette(root, train=True, transform=None, target_transform=None)

Bases: torchvision.datasets.ImageFolder

did.utils.imagenet.load_imagenet(traindir, valdir, resize_size, batch_size, num_workers, n_classes=None, shuffle=False)