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

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