__init__.py 410 Bytes
Newer Older
Kai Chen's avatar
Kai Chen committed
1
2
3
4
5
6
7
8
9
from .detectors import BaseDetector, RPN, FasterRCNN, MaskRCNN
from .builder import (build_neck, build_rpn_head, build_roi_extractor,
                      build_bbox_head, build_mask_head, build_detector)

__all__ = [
    'BaseDetector', 'RPN', 'FasterRCNN', 'MaskRCNN', 'build_backbone',
    'build_neck', 'build_rpn_head', 'build_roi_extractor', 'build_bbox_head',
    'build_mask_head', 'build_detector'
]