__init__.py 1.77 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
wuyuefeng's avatar
wuyuefeng committed
4
from .formating import Collect3D, DefaultFormatBundle, DefaultFormatBundle3D
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
14
15
16
17
18
# yapf: disable
from .transforms_3d import (AffineResize, BackgroundPointsFilter,
                            GlobalAlignment, GlobalRotScaleTrans,
                            IndoorPatchPointSample, IndoorPointSample,
                            ObjectNameFilter, ObjectNoise, ObjectRangeFilter,
                            ObjectSample, PointSample, PointShuffle,
                            PointsRangeFilter, RandomDropPointsColor,
                            RandomFlip3D, RandomJitterPoints, RandomShiftScale,
                            VoxelBasedPointSampler)
zhangwenwei's avatar
zhangwenwei committed
19
20

__all__ = [
zhangwenwei's avatar
zhangwenwei committed
21
    'ObjectSample', 'RandomFlip3D', 'ObjectNoise', 'GlobalRotScaleTrans',
22
23
24
    'PointShuffle', 'ObjectRangeFilter', 'PointsRangeFilter', 'Collect3D',
    'Compose', 'LoadMultiViewImageFromFiles', 'LoadPointsFromFile',
    'DefaultFormatBundle', 'DefaultFormatBundle3D', 'DataBaseSampler',
zhangwenwei's avatar
zhangwenwei committed
25
    'NormalizePointsColor', 'LoadAnnotations3D', 'IndoorPointSample',
26
27
28
29
    'PointSample', 'PointSegClassMapping', 'MultiScaleFlipAug3D',
    'LoadPointsFromMultiSweeps', 'BackgroundPointsFilter',
    'VoxelBasedPointSampler', 'GlobalAlignment', 'IndoorPatchPointSample',
    'LoadImageFromFileMono3D', 'ObjectNameFilter', 'RandomDropPointsColor',
30
31
    'RandomJitterPoints', 'AffineResize', 'RandomShiftScale',
    'LoadPointsFromDict'
zhangwenwei's avatar
zhangwenwei committed
32
]