__init__.py 1.27 KB
Newer Older
1
2
from mmdet.datasets.builder import build_dataloader
from .builder import DATASETS, build_dataset
zhangwenwei's avatar
zhangwenwei committed
3
from .custom_3d import Custom3DDataset
zhangwenwei's avatar
zhangwenwei committed
4
from .kitti_dataset import KittiDataset
wangtai's avatar
wangtai committed
5
from .lyft_dataset import LyftDataset
zhangwenwei's avatar
zhangwenwei committed
6
from .nuscenes_dataset import NuScenesDataset
7
8
9
10
11
12
from .pipelines import (BackgroundPointsFilter, GlobalRotScaleTrans,
                        IndoorPointSample, LoadAnnotations3D,
                        LoadPointsFromFile, LoadPointsFromMultiSweeps,
                        NormalizePointsColor, ObjectNoise, ObjectRangeFilter,
                        ObjectSample, PointShuffle, PointsRangeFilter,
                        RandomFlip3D)
13
14
from .scannet_dataset import ScanNetDataset
from .sunrgbd_dataset import SUNRGBDDataset
zhangwenwei's avatar
zhangwenwei committed
15
16
17

__all__ = [
    'KittiDataset', 'GroupSampler', 'DistributedGroupSampler',
zhangwenwei's avatar
zhangwenwei committed
18
    'build_dataloader', 'RepeatFactorDataset', 'DATASETS', 'build_dataset',
zhangwenwei's avatar
zhangwenwei committed
19
20
21
    'CocoDataset', 'NuScenesDataset', 'LyftDataset', 'ObjectSample',
    'RandomFlip3D', 'ObjectNoise', 'GlobalRotScaleTrans', 'PointShuffle',
    'ObjectRangeFilter', 'PointsRangeFilter', 'Collect3D',
zhangwenwei's avatar
zhangwenwei committed
22
    'LoadPointsFromFile', 'NormalizePointsColor', 'IndoorPointSample',
liyinhao's avatar
liyinhao committed
23
    'LoadAnnotations3D', 'SUNRGBDDataset', 'ScanNetDataset', 'Custom3DDataset',
24
    'LoadPointsFromMultiSweeps', 'BackgroundPointsFilter'
zhangwenwei's avatar
zhangwenwei committed
25
]