__init__.py 868 Bytes
Newer Older
zhangwenwei's avatar
zhangwenwei committed
1
from .base import Base3DDetector
2
from .centerpoint import CenterPoint
zhangwenwei's avatar
zhangwenwei committed
3
from .dynamic_voxelnet import DynamicVoxelNet
4
from .fcos_mono3d import FCOSMono3D
hjin2902's avatar
hjin2902 committed
5
from .groupfree3dnet import GroupFree3DNet
encore-zhou's avatar
encore-zhou committed
6
from .h3dnet import H3DNet
7
from .imvotenet import ImVoteNet
8
from .imvoxelnet import ImVoxelNet
zhangwenwei's avatar
zhangwenwei committed
9
from .mvx_faster_rcnn import DynamicMVXFasterRCNN, MVXFasterRCNN
zhangwenwei's avatar
zhangwenwei committed
10
from .mvx_two_stage import MVXTwoStageDetector
wuyuefeng's avatar
wuyuefeng committed
11
from .parta2 import PartA2
12
from .single_stage_mono3d import SingleStageMono3DDetector
13
from .ssd3dnet import SSD3DNet
wuyuefeng's avatar
Votenet  
wuyuefeng committed
14
from .votenet import VoteNet
zhangwenwei's avatar
zhangwenwei committed
15
from .voxelnet import VoxelNet
zhangwenwei's avatar
zhangwenwei committed
16
17

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