__init__.py 1.03 KB
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
Wenbo Yu's avatar
Wenbo Yu committed
14
from .sassd import SASSD
15
from .single_stage_mono3d import SingleStageMono3DDetector
16
from .smoke_mono3d import SMOKEMono3D
17
from .ssd3dnet import SSD3DNet
wuyuefeng's avatar
Votenet  
wuyuefeng committed
18
from .votenet import VoteNet
zhangwenwei's avatar
zhangwenwei committed
19
from .voxelnet import VoxelNet
zhangwenwei's avatar
zhangwenwei committed
20
21

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