"src/targets/vscode:/vscode.git/clone" did not exist on "d7a829f4c7538fc52e0d95cd3d1518ab4a639c8e"
__init__.py 1.94 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
from .pipelines import (BackgroundPointsFilter, GlobalAlignment,
                        GlobalRotScaleTrans, IndoorPatchPointSample,
12
13
14
15
                        IndoorPointSample, LoadAnnotations3D,
                        LoadPointsFromFile, LoadPointsFromMultiSweeps,
                        NormalizePointsColor, ObjectNoise, ObjectRangeFilter,
                        ObjectSample, PointShuffle, PointsRangeFilter,
16
                        RandomFlip3D, VoxelBasedPointSampler)
17
from .s3dis_dataset import S3DISSegDataset
18
from .scannet_dataset import ScanNetDataset, ScanNetSegDataset
19
from .semantickitti_dataset import SemanticKITTIDataset
20
from .sunrgbd_dataset import SUNRGBDDataset
21
from .utils import get_loading_pipeline
Wenwei Zhang's avatar
Wenwei Zhang committed
22
from .waymo_dataset import WaymoDataset
zhangwenwei's avatar
zhangwenwei committed
23
24

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