mmpretrain/mmcls/datasets/__init__.py

from .base_dataset import BaseDataset
from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
from .cifar import CIFAR10, CIFAR100
from .dataset_wrappers import (ClassBalancedDataset, ConcatDataset,
                               RepeatDataset)
from .imagenet import ImageNet
from .mnist import MNIST, FashionMNIST
from .samplers import DistributedSampler

__all__ = [
    'BaseDataset', 'ImageNet', 'CIFAR10', 'CIFAR100', 'MNIST', 'FashionMNIST',
    'build_dataloader', 'build_dataset', 'Compose', 'DistributedSampler',
    'ConcatDataset', 'RepeatDataset', 'ClassBalancedDataset', 'DATASETS',
    'PIPELINES'
]