27 lines
1.3 KiB
Python
27 lines
1.3 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
|
|
from .cityscapes import CityscapesDataset
|
|
from .coco import CocoDataset
|
|
from .coco_panoptic import CocoPanopticDataset
|
|
from .custom import CustomDataset
|
|
from .dataset_wrappers import (ClassBalancedDataset, ConcatDataset,
|
|
MultiImageMixDataset, RepeatDataset)
|
|
from .deepfashion import DeepFashionDataset
|
|
from .lvis import LVISDataset, LVISV1Dataset, LVISV05Dataset
|
|
from .samplers import DistributedGroupSampler, DistributedSampler, GroupSampler
|
|
from .utils import (NumClassCheckHook, get_loading_pipeline,
|
|
replace_ImageToTensor)
|
|
from .voc import VOCDataset
|
|
from .wider_face import WIDERFaceDataset
|
|
from .xml_style import XMLDataset
|
|
|
|
__all__ = [
|
|
'CustomDataset', 'XMLDataset', 'CocoDataset', 'DeepFashionDataset',
|
|
'VOCDataset', 'CityscapesDataset', 'LVISDataset', 'LVISV05Dataset',
|
|
'LVISV1Dataset', 'GroupSampler', 'DistributedGroupSampler',
|
|
'DistributedSampler', 'build_dataloader', 'ConcatDataset', 'RepeatDataset',
|
|
'ClassBalancedDataset', 'WIDERFaceDataset', 'DATASETS', 'PIPELINES',
|
|
'build_dataset', 'replace_ImageToTensor', 'get_loading_pipeline',
|
|
'NumClassCheckHook', 'CocoPanopticDataset', 'MultiImageMixDataset'
|
|
]
|