__init__.py 1016 Bytes
Newer Older
dingchang's avatar
dingchang committed
1
# Copyright (c) OpenMMLab. All rights reserved.
zhangwenwei's avatar
zhangwenwei committed
2
from .base import Base3DDetector
3
from .centerpoint import CenterPoint
zhangwenwei's avatar
zhangwenwei committed
4
from .dynamic_voxelnet import DynamicVoxelNet
5
from .fcos_mono3d import FCOSMono3D
hjin2902's avatar
hjin2902 committed
6
from .groupfree3dnet import GroupFree3DNet
encore-zhou's avatar
encore-zhou committed
7
from .h3dnet import H3DNet
8
from .imvotenet import ImVoteNet
9
from .imvoxelnet import ImVoxelNet
zhangwenwei's avatar
zhangwenwei committed
10
from .mvx_faster_rcnn import DynamicMVXFasterRCNN, MVXFasterRCNN
zhangwenwei's avatar
zhangwenwei committed
11
from .mvx_two_stage import MVXTwoStageDetector
wuyuefeng's avatar
wuyuefeng committed
12
from .parta2 import PartA2
13
from .point_rcnn import PointRCNN
14
from .single_stage_mono3d import SingleStageMono3DDetector
15
from .smoke_mono3d import SMOKEMono3D
16
from .ssd3dnet import SSD3DNet
wuyuefeng's avatar
Votenet  
wuyuefeng committed
17
from .votenet import VoteNet
zhangwenwei's avatar
zhangwenwei committed
18
from .voxelnet import VoxelNet
zhangwenwei's avatar
zhangwenwei committed
19
20

__all__ = [
21
22
23
    'Base3DDetector', 'VoxelNet', 'DynamicVoxelNet', 'MVXTwoStageDetector',
    'DynamicMVXFasterRCNN', 'MVXFasterRCNN', 'PartA2', 'VoteNet', 'H3DNet',
    'CenterPoint', 'SSD3DNet', 'ImVoteNet', 'SingleStageMono3DDetector',
24
    'FCOSMono3D', 'ImVoxelNet', 'GroupFree3DNet', 'PointRCNN', 'SMOKEMono3D'
zhangwenwei's avatar
zhangwenwei committed
25
]