__init__.py 2.08 KB
Newer Older
dingchang's avatar
dingchang committed
1
# Copyright (c) OpenMMLab. All rights reserved.
zhangwenwei's avatar
zhangwenwei committed
2
from .dbsampler import DataBaseSampler
jshilong's avatar
jshilong committed
3
from .formating import Pack3DDetInputs
4
from .loading import (LoadAnnotations3D, LoadImageFromFileMono3D,
5
6
                      LoadMultiViewImageFromFiles, LoadPointsFromDict,
                      LoadPointsFromFile, LoadPointsFromMultiSweeps,
7
8
                      MonoDet3DInferencerLoader,
                      MultiModalityDet3DInferencerLoader, NormalizePointsColor,
9
                      PointSegClassMapping)
zhangwenwei's avatar
zhangwenwei committed
10
from .test_time_aug import MultiScaleFlipAug3D
11
12
13
14
# yapf: disable
from .transforms_3d import (AffineResize, BackgroundPointsFilter,
                            GlobalAlignment, GlobalRotScaleTrans,
                            IndoorPatchPointSample, IndoorPointSample,
15
16
                            LaserMix, MultiViewWrapper, ObjectNameFilter,
                            ObjectNoise, ObjectRangeFilter, ObjectSample,
17
                            PhotoMetricDistortion3D, PointSample, PointShuffle,
18
                            PointsRangeFilter, PolarMix, RandomDropPointsColor,
19
20
                            RandomFlip3D, RandomJitterPoints, RandomResize3D,
                            RandomShiftScale, Resize3D, VoxelBasedPointSampler)
zhangwenwei's avatar
zhangwenwei committed
21
22

__all__ = [
zhangwenwei's avatar
zhangwenwei committed
23
    'ObjectSample', 'RandomFlip3D', 'ObjectNoise', 'GlobalRotScaleTrans',
jshilong's avatar
jshilong committed
24
    'PointShuffle', 'ObjectRangeFilter', 'PointsRangeFilter',
25
26
27
28
29
30
31
32
    'Pack3DDetInputs', 'LoadMultiViewImageFromFiles', 'LoadPointsFromFile',
    'DataBaseSampler', 'NormalizePointsColor', 'LoadAnnotations3D',
    'IndoorPointSample', 'PointSample', 'PointSegClassMapping',
    'MultiScaleFlipAug3D', 'LoadPointsFromMultiSweeps',
    'BackgroundPointsFilter', 'VoxelBasedPointSampler', 'GlobalAlignment',
    'IndoorPatchPointSample', 'LoadImageFromFileMono3D', 'ObjectNameFilter',
    'RandomDropPointsColor', 'RandomJitterPoints', 'AffineResize',
    'RandomShiftScale', 'LoadPointsFromDict', 'Resize3D', 'RandomResize3D',
33
    'MultiViewWrapper', 'PhotoMetricDistortion3D', 'MonoDet3DInferencerLoader',
34
35
    'LidarDet3DInferencerLoader', 'PolarMix', 'LaserMix',
    'MultiModalityDet3DInferencerLoader'
zhangwenwei's avatar
zhangwenwei committed
36
]