__init__.py 1.25 KB
Newer Older
1
from mmdet.datasets.builder import DATASETS
zhangwenwei's avatar
zhangwenwei committed
2
from .builder import build_dataset
zhangwenwei's avatar
zhangwenwei committed
3
from .dataset_wrappers import RepeatFactorDataset
zhangwenwei's avatar
zhangwenwei committed
4
5
6
7
from .kitti2d_dataset import Kitti2DDataset
from .kitti_dataset import KittiDataset
from .loader import DistributedGroupSampler, GroupSampler, build_dataloader
from .nuscenes_dataset import NuScenesDataset
liyinhao's avatar
liyinhao committed
8
from .pipelines import (GlobalRotScale, IndoorFlipData, IndoorGlobalRotScale,
liyinhao's avatar
liyinhao committed
9
10
                        IndoorLoadAnnotations3D, IndoorLoadPointsFromFile,
                        IndoorPointsColorJitter, IndoorPointsColorNormalize,
liyinhao's avatar
liyinhao committed
11
12
                        ObjectNoise, ObjectRangeFilter, ObjectSample,
                        PointSample, PointShuffle, PointsRangeFilter,
13
                        RandomFlip3D)
zhangwenwei's avatar
zhangwenwei committed
14
15
16

__all__ = [
    'KittiDataset', 'GroupSampler', 'DistributedGroupSampler',
zhangwenwei's avatar
zhangwenwei committed
17
    'build_dataloader', 'RepeatFactorDataset', 'DATASETS', 'build_dataset',
18
19
    'CocoDataset', 'Kitti2DDataset', 'NuScenesDataset', 'ObjectSample',
    'RandomFlip3D', 'ObjectNoise', 'GlobalRotScale', 'PointShuffle',
liyinhao's avatar
liyinhao committed
20
    'ObjectRangeFilter', 'PointsRangeFilter', 'Collect3D',
liyinhao's avatar
liyinhao committed
21
    'IndoorLoadPointsFromFile', 'IndoorPointsColorNormalize', 'PointSample',
liyinhao's avatar
liyinhao committed
22
23
    'IndoorLoadAnnotations3D', 'IndoorPointsColorJitter',
    'IndoorGlobalRotScale', 'IndoorFlipData'
zhangwenwei's avatar
zhangwenwei committed
24
]