__init__.py 1.08 KB
Newer Older
zhangwenwei's avatar
zhangwenwei committed
1
from .backbones import *  # noqa: F401,F403
2
3
from .builder import (FUSION_LAYERS, MIDDLE_ENCODERS, VOXEL_ENCODERS,
                      build_backbone, build_detector, build_fusion_layer,
zhangwenwei's avatar
zhangwenwei committed
4
5
6
                      build_head, build_loss, build_middle_encoder, build_neck,
                      build_roi_extractor, build_shared_head,
                      build_voxel_encoder)
7
from .decode_heads import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
8
from .dense_heads import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
9
10
11
12
from .detectors import *  # noqa: F401,F403
from .fusion_layers import *  # noqa: F401,F403
from .losses import *  # noqa: F401,F403
from .middle_encoders import *  # noqa: F401,F403
wuyuefeng's avatar
Votenet  
wuyuefeng committed
13
from .model_utils import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
14
from .necks import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
15
from .roi_heads import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
16
17
18
from .voxel_encoders import *  # noqa: F401,F403

__all__ = [
zhangwenwei's avatar
zhangwenwei committed
19
    'VOXEL_ENCODERS', 'MIDDLE_ENCODERS', 'FUSION_LAYERS', 'build_backbone',
zhangwenwei's avatar
zhangwenwei committed
20
    'build_neck', 'build_roi_extractor', 'build_shared_head', 'build_head',
zhangwenwei's avatar
zhangwenwei committed
21
22
    'build_loss', 'build_detector', 'build_fusion_layer',
    'build_middle_encoder', 'build_voxel_encoder'
zhangwenwei's avatar
zhangwenwei committed
23
]