ares.dataset package

ares.dataset.cifar_dataset.cifar10(batchsize, cifar10_path)[source]

The function to create cifar10 dataloader.

class ares.dataset.imagenet_dataset.ImageNetDataset(data_dir, meta_file, transform=None)[source]

Bases: Dataset

The class to create ImageNet dataset.

__init__(data_dir, meta_file, transform=None)[source]

The function to initialize ImageNet class.

Parameters:
  • data_dir (str) – The path to the dataset.

  • meta_file (str) – The path to the file containing image directories and labels.

  • transform (torchvision.transforms) – The transform for input image.