__init__.py 1.34 KB
Newer Older
1
from mmdet.datasets.pipelines import Compose
zhangwenwei's avatar
zhangwenwei committed
2
from .dbsampler import DataBaseSampler
wuyuefeng's avatar
wuyuefeng committed
3
from .formating import Collect3D, DefaultFormatBundle, DefaultFormatBundle3D
4
5
6
7
from .loading import (LoadAnnotations3D, LoadImageFromFileMono3D,
                      LoadMultiViewImageFromFiles, LoadPointsFromFile,
                      LoadPointsFromMultiSweeps, NormalizePointsColor,
                      PointSegClassMapping)
zhangwenwei's avatar
zhangwenwei committed
8
from .test_time_aug import MultiScaleFlipAug3D
9
from .transforms_3d import (BackgroundPointsFilter, GlobalRotScaleTrans,
10
11
12
13
                            IndoorPatchPointSample, IndoorPointSample,
                            ObjectNoise, ObjectRangeFilter, ObjectSample,
                            PointShuffle, PointsRangeFilter, RandomFlip3D,
                            VoxelBasedPointSampler)
zhangwenwei's avatar
zhangwenwei committed
14
15

__all__ = [
zhangwenwei's avatar
zhangwenwei committed
16
    'ObjectSample', 'RandomFlip3D', 'ObjectNoise', 'GlobalRotScaleTrans',
17
18
19
    'PointShuffle', 'ObjectRangeFilter', 'PointsRangeFilter', 'Collect3D',
    'Compose', 'LoadMultiViewImageFromFiles', 'LoadPointsFromFile',
    'DefaultFormatBundle', 'DefaultFormatBundle3D', 'DataBaseSampler',
zhangwenwei's avatar
zhangwenwei committed
20
    'NormalizePointsColor', 'LoadAnnotations3D', 'IndoorPointSample',
21
    'PointSegClassMapping', 'MultiScaleFlipAug3D', 'LoadPointsFromMultiSweeps',
22
    'BackgroundPointsFilter', 'VoxelBasedPointSampler',
23
    'IndoorPatchPointSample', 'LoadImageFromFileMono3D'
zhangwenwei's avatar
zhangwenwei committed
24
]