mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
2ce5825ef1
* add imagnet21k * Update unit test * update imaenet21k * use slots * use slots * rename Data_item to ImageInfo * add unit tests * Update unit tests * rm some print * update unit tests * fix lint * remove default value of pipeline
20 lines
813 B
Python
20 lines
813 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
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 .imagenet21k import ImageNet21k
|
|
from .mnist import MNIST, FashionMNIST
|
|
from .multi_label import MultiLabelDataset
|
|
from .samplers import DistributedSampler
|
|
from .voc import VOC
|
|
|
|
__all__ = [
|
|
'BaseDataset', 'ImageNet', 'CIFAR10', 'CIFAR100', 'MNIST', 'FashionMNIST',
|
|
'VOC', 'MultiLabelDataset', 'build_dataloader', 'build_dataset',
|
|
'DistributedSampler', 'ConcatDataset', 'RepeatDataset',
|
|
'ClassBalancedDataset', 'DATASETS', 'PIPELINES', 'ImageNet21k'
|
|
]
|