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

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