__init__.py 1.46 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
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
8
9
10
11
from .pipelines import (BackgroundPointsFilter, GlobalRotScaleTrans,
                        IndoorPointSample, LoadAnnotations3D,
                        LoadPointsFromFile, LoadPointsFromMultiSweeps,
                        NormalizePointsColor, ObjectNoise, ObjectRangeFilter,
                        ObjectSample, PointShuffle, PointsRangeFilter,
12
                        RandomFlip3D, VoxelBasedPointSampler)
13
from .scannet_dataset import ScanNetDataset
14
from .semantickitti_dataset import SemanticKITTIDataset
15
from .sunrgbd_dataset import SUNRGBDDataset
Wenwei Zhang's avatar
Wenwei Zhang committed
16
from .waymo_dataset import WaymoDataset
zhangwenwei's avatar
zhangwenwei committed
17
18
19

__all__ = [
    'KittiDataset', 'GroupSampler', 'DistributedGroupSampler',
zhangwenwei's avatar
zhangwenwei committed
20
    'build_dataloader', 'RepeatFactorDataset', 'DATASETS', 'build_dataset',
zhangwenwei's avatar
zhangwenwei committed
21
22
23
    'CocoDataset', 'NuScenesDataset', 'LyftDataset', 'ObjectSample',
    'RandomFlip3D', 'ObjectNoise', 'GlobalRotScaleTrans', 'PointShuffle',
    'ObjectRangeFilter', 'PointsRangeFilter', 'Collect3D',
zhangwenwei's avatar
zhangwenwei committed
24
    'LoadPointsFromFile', 'NormalizePointsColor', 'IndoorPointSample',
25
26
27
    'LoadAnnotations3D', 'SUNRGBDDataset', 'ScanNetDataset',
    'SemanticKITTIDataset', 'Custom3DDataset', 'LoadPointsFromMultiSweeps',
    'WaymoDataset', 'BackgroundPointsFilter', 'VoxelBasedPointSampler'
zhangwenwei's avatar
zhangwenwei committed
28
]