"sgl-kernel/vscode:/vscode.git/clone" did not exist on "3c9740d2004fff7801433efa88e2dcfdd828bc0a"
__init__.py 2.33 KB
Newer Older
dingchang's avatar
dingchang committed
1
# Copyright (c) OpenMMLab. All rights reserved.
2
from .builder import DATASETS, PIPELINES, build_dataset
3
from .custom_3d_seg import Custom3DSegDataset
jshilong's avatar
jshilong committed
4
from .det3d_dataset import Det3DDataset
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
# yapf: disable
11
from .pipelines import (AffineResize, BackgroundPointsFilter, GlobalAlignment,
12
                        GlobalRotScaleTrans, IndoorPatchPointSample,
13
                        IndoorPointSample, LoadAnnotations3D,
14
15
16
17
18
19
                        LoadPointsFromDict, LoadPointsFromFile,
                        LoadPointsFromMultiSweeps, NormalizePointsColor,
                        ObjectNameFilter, ObjectNoise, ObjectRangeFilter,
                        ObjectSample, PointSample, PointShuffle,
                        PointsRangeFilter, RandomDropPointsColor, RandomFlip3D,
                        RandomJitterPoints, RandomShiftScale,
20
21
                        VoxelBasedPointSampler)
# yapf: enable
22
from .s3dis_dataset import S3DISDataset, S3DISSegDataset
23
24
from .scannet_dataset import (ScanNetDataset, ScanNetInstanceSegDataset,
                              ScanNetSegDataset)
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',
jshilong's avatar
jshilong committed
39
    'Det3DDataset', 'Custom3DSegDataset', 'LoadPointsFromMultiSweeps',
40
41
42
43
    'WaymoDataset', 'BackgroundPointsFilter', 'VoxelBasedPointSampler',
    'get_loading_pipeline', 'RandomDropPointsColor', 'RandomJitterPoints',
    'ObjectNameFilter', 'AffineResize', 'RandomShiftScale',
    'LoadPointsFromDict', 'PIPELINES'
zhangwenwei's avatar
zhangwenwei committed
44
]