__init__.py 1.43 KB
Newer Older
dingchang's avatar
dingchang committed
1
# Copyright (c) OpenMMLab. All rights reserved.
zhangwenwei's avatar
zhangwenwei committed
2
from .backbones import *  # noqa: F401,F403
3
4
5
6
7
8
9
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)
10
from .decode_heads import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
11
from .dense_heads import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
12
13
14
15
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
16
from .model_utils import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
17
from .necks import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
18
from .roi_heads import *  # noqa: F401,F403
19
from .segmentors import *  # noqa: F401,F403
zhangwenwei's avatar
zhangwenwei committed
20
21
22
from .voxel_encoders import *  # noqa: F401,F403

__all__ = [
23
24
25
26
27
28
    '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
29
]