__init__.py 1.71 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
4
from .custom_3d_seg import Custom3DSegDataset
zhangwenwei's avatar
zhangwenwei committed
5
from .kitti_dataset import KittiDataset
wangtai's avatar
wangtai committed
6
from .lyft_dataset import LyftDataset
zhangwenwei's avatar
zhangwenwei committed
7
from .nuscenes_dataset import NuScenesDataset
8
from .nuscenes_mono_dataset import NuScenesMonoDataset
9
10
11
12
13
from .pipelines import (BackgroundPointsFilter, GlobalRotScaleTrans,
                        IndoorPointSample, LoadAnnotations3D,
                        LoadPointsFromFile, LoadPointsFromMultiSweeps,
                        NormalizePointsColor, ObjectNoise, ObjectRangeFilter,
                        ObjectSample, PointShuffle, PointsRangeFilter,
14
                        RandomFlip3D, VoxelBasedPointSampler)
15
from .scannet_dataset import ScanNetDataset, ScanNetSegDataset
16
from .semantickitti_dataset import SemanticKITTIDataset
17
from .sunrgbd_dataset import SUNRGBDDataset
18
from .utils import get_loading_pipeline
Wenwei Zhang's avatar
Wenwei Zhang committed
19
from .waymo_dataset import WaymoDataset
zhangwenwei's avatar
zhangwenwei committed
20
21
22

__all__ = [
    'KittiDataset', 'GroupSampler', 'DistributedGroupSampler',
zhangwenwei's avatar
zhangwenwei committed
23
    'build_dataloader', 'RepeatFactorDataset', 'DATASETS', 'build_dataset',
24
25
26
    'CocoDataset', 'NuScenesDataset', 'NuScenesMonoDataset', 'LyftDataset',
    'ObjectSample', 'RandomFlip3D', 'ObjectNoise', 'GlobalRotScaleTrans',
    'PointShuffle', 'ObjectRangeFilter', 'PointsRangeFilter', 'Collect3D',
zhangwenwei's avatar
zhangwenwei committed
27
    'LoadPointsFromFile', 'NormalizePointsColor', 'IndoorPointSample',
28
    'LoadAnnotations3D', 'SUNRGBDDataset', 'ScanNetDataset',
29
30
    'ScanNetSegDataset', 'SemanticKITTIDataset', 'Custom3DDataset',
    'Custom3DSegDataset', 'LoadPointsFromMultiSweeps', 'WaymoDataset',
31
    'BackgroundPointsFilter', 'VoxelBasedPointSampler', 'get_loading_pipeline'
zhangwenwei's avatar
zhangwenwei committed
32
]