walt/datasets/__init__.py

from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
from mmdet.datasets.cityscapes import CityscapesDataset
from mmdet.datasets.coco import CocoDataset
from .custom import CustomDatasetLocal
from mmdet.datasets.custom import CustomDataset
from mmdet.datasets.dataset_wrappers import (ClassBalancedDataset, ConcatDataset,
                               RepeatDataset)
from mmdet.datasets.deepfashion import DeepFashionDataset
from mmdet.datasets.lvis import LVISDataset, LVISV1Dataset, LVISV05Dataset
from mmdet.datasets.samplers import DistributedGroupSampler, DistributedSampler, GroupSampler
from mmdet.datasets.utils import (NumClassCheckHook, get_loading_pipeline,
                    replace_ImageToTensor)
from mmdet.datasets.voc import VOCDataset
from mmdet.datasets.wider_face import WIDERFaceDataset
from mmdet.datasets.xml_style import XMLDataset
from .walt_synthetic import WaltSynthDataset
from .walt_3d import Walt3DDataset
from .walt import WaltDataset
__all__ = [
    'CustomDataset', 'XMLDataset', 'CocoDataset', 'DeepFashionDataset',
    'VOCDataset', 'CityscapesDataset', 'LVISDataset', 'LVISV05Dataset',
    'LVISV1Dataset', 'GroupSampler', 'DistributedGroupSampler',
    'DistributedSampler', 'build_dataloader', 'ConcatDataset', 'RepeatDataset',
    'ClassBalancedDataset', 'Walt3DDataset','WIDERFaceDataset', 'DATASETS', 'PIPELINES',
    'build_dataset', 'replace_ImageToTensor', 'get_loading_pipeline',
    'WaltSynthDataset',  'WaltDataset',  'NumClassCheckHook'
]