__init__.py 1.2 KB
Newer Older
yinchimaoliang's avatar
yinchimaoliang committed
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
wuyuefeng's avatar
wuyuefeng committed
7
8
from .pipelines import (GlobalRotScaleTrans, IndoorPointSample,
                        LoadAnnotations3D, LoadPointsFromFile,
liyinhao's avatar
liyinhao committed
9
10
11
                        LoadPointsFromMultiSweeps, NormalizePointsColor,
                        ObjectNoise, ObjectRangeFilter, ObjectSample,
                        PointShuffle, PointsRangeFilter, RandomFlip3D)
12
13
from .scannet_dataset import ScanNetDataset
from .sunrgbd_dataset import SUNRGBDDataset
zhangwenwei's avatar
zhangwenwei committed
14
15
16

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