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

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