__init__.py 1.58 KB
Newer Older
dingchang's avatar
dingchang committed
1
# Copyright (c) OpenMMLab. All rights reserved.
zhangshilong's avatar
zhangshilong committed
2
from mmdet3d.models.layers.fusion_layers import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
3
from .backbones import *  # noqa: F401,F403
4
5
6
7
8
9
10
from .builder import (BACKBONES, DETECTORS, FUSION_LAYERS, HEADS, LOSSES,
                      MIDDLE_ENCODERS, NECKS, ROI_EXTRACTORS, SEGMENTORS,
                      SHARED_HEADS, VOXEL_ENCODERS, build_backbone,
                      build_detector, build_fusion_layer, build_head,
                      build_loss, build_middle_encoder, build_model,
                      build_neck, build_roi_extractor, build_shared_head,
                      build_voxel_encoder)
11
from .data_preprocessors import *  # noqa: F401,F403
12
from .decode_heads import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
13
from .dense_heads import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
14
from .detectors import *  # noqa: F401,F403
zhangshilong's avatar
zhangshilong committed
15
from .layers import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
16
17
18
from .losses import *  # noqa: F401,F403
from .middle_encoders import *  # noqa: F401,F403
from .necks import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
19
from .roi_heads import *  # noqa: F401,F403
20
from .segmentors import *  # noqa: F401,F403
zhangshilong's avatar
zhangshilong committed
21
22
from .test_time_augs import *  # noqa: F401,F403
from .utils import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
23
24
25
from .voxel_encoders import *  # noqa: F401,F403

__all__ = [
26
27
28
29
30
31
    'BACKBONES', 'NECKS', 'ROI_EXTRACTORS', 'SHARED_HEADS', 'HEADS', 'LOSSES',
    'DETECTORS', 'SEGMENTORS', 'VOXEL_ENCODERS', 'MIDDLE_ENCODERS',
    'FUSION_LAYERS', 'build_backbone', 'build_neck', 'build_roi_extractor',
    'build_shared_head', 'build_head', 'build_loss', 'build_detector',
    'build_fusion_layer', 'build_model', 'build_middle_encoder',
    'build_voxel_encoder'
zhangwenwei's avatar
zhangwenwei committed
32
]