__init__.py 683 Bytes
Newer Older
Kai Chen's avatar
Kai Chen committed
1
2
3
4
5
6
7
from .backbones import *  # noqa: F401,F403
from .necks import *  # noqa: F401,F403
from .roi_extractors import *  # noqa: F401,F403
from .anchor_heads import *  # noqa: F401,F403
from .bbox_heads import *  # noqa: F401,F403
from .mask_heads import *  # noqa: F401,F403
from .detectors import *  # noqa: F401,F403
Kai Chen's avatar
Kai Chen committed
8
9
10
from .registry import BACKBONES, NECKS, ROI_EXTRACTORS, HEADS, DETECTORS
from .builder import (build_backbone, build_neck, build_roi_extractor,
                      build_head, build_detector)
Kai Chen's avatar
Kai Chen committed
11
12

__all__ = [
Kai Chen's avatar
Kai Chen committed
13
14
    'BACKBONES', 'NECKS', 'ROI_EXTRACTORS', 'HEADS', 'DETECTORS',
    'build_backbone', 'build_neck', 'build_roi_extractor', 'build_head',
15
    'build_detector'
Kai Chen's avatar
Kai Chen committed
16
]