__init__.py 916 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 .single_stage_mono3d import SingleStageMono3DDetector
14
from .ssd3dnet import SSD3DNet
wuyuefeng's avatar
Votenet  
wuyuefeng committed
15
from .votenet import VoteNet
zhangwenwei's avatar
zhangwenwei committed
16
from .voxelnet import VoxelNet
zhangwenwei's avatar
zhangwenwei committed
17
18

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