__init__.py 932 Bytes
Newer Older
Kai Chen's avatar
Kai Chen committed
1
2
3
4
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
myownskyW7's avatar
myownskyW7 committed
5
from .shared_heads import *  # noqa: F401,F403
Kai Chen's avatar
Kai Chen committed
6
7
from .bbox_heads import *  # noqa: F401,F403
from .mask_heads import *  # noqa: F401,F403
Jiangmiao Pang's avatar
Jiangmiao Pang committed
8
from .losses import *  # noqa: F401,F403
Kai Chen's avatar
Kai Chen committed
9
from .detectors import *  # noqa: F401,F403
myownskyW7's avatar
myownskyW7 committed
10
from .registry import (BACKBONES, NECKS, ROI_EXTRACTORS, SHARED_HEADS, HEADS,
Jiangmiao Pang's avatar
Jiangmiao Pang committed
11
                       LOSSES, DETECTORS)
Kai Chen's avatar
Kai Chen committed
12
from .builder import (build_backbone, build_neck, build_roi_extractor,
Jiangmiao Pang's avatar
Jiangmiao Pang committed
13
14
                      build_shared_head, build_head, build_loss,
                      build_detector)
Kai Chen's avatar
Kai Chen committed
15
16

__all__ = [
Jiangmiao Pang's avatar
Jiangmiao Pang committed
17
    'BACKBONES', 'NECKS', 'ROI_EXTRACTORS', 'SHARED_HEADS', 'HEADS', 'LOSSES',
myownskyW7's avatar
myownskyW7 committed
18
    'DETECTORS', 'build_backbone', 'build_neck', 'build_roi_extractor',
Jiangmiao Pang's avatar
Jiangmiao Pang committed
19
    'build_shared_head', 'build_head', 'build_loss', 'build_detector'
Kai Chen's avatar
Kai Chen committed
20
]