__init__.py 2.32 KB
Newer Older
dingchang's avatar
dingchang committed
1
# Copyright (c) OpenMMLab. All rights reserved.
2
from .builder import DATASETS, PIPELINES, build_dataset
jshilong's avatar
jshilong committed
3
from .det3d_dataset import Det3DDataset
zhangwenwei's avatar
zhangwenwei committed
4
from .kitti_dataset import KittiDataset
5
from .kitti_mono_dataset import KittiMonoDataset
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
# yapf: disable
10
from .pipelines import (AffineResize, BackgroundPointsFilter, GlobalAlignment,
11
                        GlobalRotScaleTrans, IndoorPatchPointSample,
12
                        IndoorPointSample, LoadAnnotations3D,
13
14
15
16
17
18
                        LoadPointsFromDict, LoadPointsFromFile,
                        LoadPointsFromMultiSweeps, NormalizePointsColor,
                        ObjectNameFilter, ObjectNoise, ObjectRangeFilter,
                        ObjectSample, PointSample, PointShuffle,
                        PointsRangeFilter, RandomDropPointsColor, RandomFlip3D,
                        RandomJitterPoints, RandomShiftScale,
19
20
                        VoxelBasedPointSampler)
# yapf: enable
21
from .s3dis_dataset import S3DISDataset, S3DISSegDataset
22
23
from .scannet_dataset import (ScanNetDataset, ScanNetInstanceSegDataset,
                              ScanNetSegDataset)
ZCMax's avatar
ZCMax committed
24
from .seg3d_dataset import Seg3DDataset
25
from .semantickitti_dataset import SemanticKITTIDataset
26
from .sunrgbd_dataset import SUNRGBDDataset
27
from .utils import get_loading_pipeline
Wenwei Zhang's avatar
Wenwei Zhang committed
28
from .waymo_dataset import WaymoDataset
zhangwenwei's avatar
zhangwenwei committed
29
30

__all__ = [
31
32
33
34
35
36
37
38
    'KittiDataset', 'KittiMonoDataset', 'DATASETS', 'build_dataset',
    'NuScenesDataset', 'NuScenesMonoDataset', 'LyftDataset', 'ObjectSample',
    'RandomFlip3D', 'ObjectNoise', 'GlobalRotScaleTrans', 'PointShuffle',
    'ObjectRangeFilter', 'PointsRangeFilter', 'LoadPointsFromFile',
    'S3DISSegDataset', 'S3DISDataset', 'NormalizePointsColor',
    'IndoorPatchPointSample', 'IndoorPointSample', 'PointSample',
    'LoadAnnotations3D', 'GlobalAlignment', 'SUNRGBDDataset', 'ScanNetDataset',
    'ScanNetSegDataset', 'ScanNetInstanceSegDataset', 'SemanticKITTIDataset',
ZCMax's avatar
ZCMax committed
39
    'Det3DDataset', 'Seg3DDataset', 'LoadPointsFromMultiSweeps',
40
41
42
43
    'WaymoDataset', 'BackgroundPointsFilter', 'VoxelBasedPointSampler',
    'get_loading_pipeline', 'RandomDropPointsColor', 'RandomJitterPoints',
    'ObjectNameFilter', 'AffineResize', 'RandomShiftScale',
    'LoadPointsFromDict', 'PIPELINES'
zhangwenwei's avatar
zhangwenwei committed
44
]