__init__.py 1014 Bytes
Newer Older
1
from .atss import ATSS
Kai Chen's avatar
Kai Chen committed
2
from .base import BaseDetector
Kai Chen's avatar
Kai Chen committed
3
from .cascade_rcnn import CascadeRCNN
4
from .double_head_rcnn import DoubleHeadRCNN
5
6
from .fast_rcnn import FastRCNN
from .faster_rcnn import FasterRCNN
7
from .fcos import FCOS
Tao Kong's avatar
Tao Kong committed
8
from .fovea import FOVEA
luxiin's avatar
luxiin committed
9
from .grid_rcnn import GridRCNN
10
11
from .htc import HybridTaskCascade
from .mask_rcnn import MaskRCNN
12
from .mask_scoring_rcnn import MaskScoringRCNN
13
from .reppoints_detector import RepPointsDetector
14
15
16
from .retinanet import RetinaNet
from .rpn import RPN
from .single_stage import SingleStageDetector
WXinlong's avatar
WXinlong committed
17
from .single_stage_ins import SingleStageInsDetector
18
from .two_stage import TwoStageDetector
WXinlong's avatar
WXinlong committed
19
from .solo import SOLO
WXinlong's avatar
WXinlong committed
20
from .solov2 import SOLOv2
Kai Chen's avatar
Kai Chen committed
21

22
__all__ = [
23
    'ATSS', 'BaseDetector', 'SingleStageDetector', 'TwoStageDetector', 'RPN',
24
    'FastRCNN', 'FasterRCNN', 'MaskRCNN', 'CascadeRCNN', 'HybridTaskCascade',
25
    'DoubleHeadRCNN', 'RetinaNet', 'FCOS', 'GridRCNN', 'MaskScoringRCNN',
WXinlong's avatar
WXinlong committed
26
    'RepPointsDetector', 'FOVEA', 'SingleStageInsDetector', 'SOLO', 'SOLOv2'
27
]