OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.9 KiB
50 lines
1.9 KiB
# Copyright (c) OpenMMLab. All rights reserved. |
|
from .atss import ATSS |
|
from .autoassign import AutoAssign |
|
from .base import BaseDetector |
|
from .cascade_rcnn import CascadeRCNN |
|
from .centernet import CenterNet |
|
from .cornernet import CornerNet |
|
from .deformable_detr import DeformableDETR |
|
from .detr import DETR |
|
from .fast_rcnn import FastRCNN |
|
from .faster_rcnn import FasterRCNN |
|
from .fcos import FCOS |
|
from .fovea import FOVEA |
|
from .fsaf import FSAF |
|
from .gfl import GFL |
|
from .grid_rcnn import GridRCNN |
|
from .htc import HybridTaskCascade |
|
from .kd_one_stage import KnowledgeDistillationSingleStageDetector |
|
from .mask_rcnn import MaskRCNN |
|
from .mask_scoring_rcnn import MaskScoringRCNN |
|
from .nasfcos import NASFCOS |
|
from .paa import PAA |
|
from .panoptic_fpn import PanopticFPN |
|
from .panoptic_two_stage_segmentor import TwoStagePanopticSegmentor |
|
from .point_rend import PointRend |
|
from .reppoints_detector import RepPointsDetector |
|
from .retinanet import RetinaNet |
|
from .rpn import RPN |
|
from .scnet import SCNet |
|
from .single_stage import SingleStageDetector |
|
from .solo import SOLO |
|
from .sparse_rcnn import SparseRCNN |
|
from .trident_faster_rcnn import TridentFasterRCNN |
|
from .two_stage import TwoStageDetector |
|
from .vfnet import VFNet |
|
from .yolact import YOLACT |
|
from .yolo import YOLOV3 |
|
from .yolof import YOLOF |
|
from .yolox import YOLOX |
|
|
|
__all__ = [ |
|
'ATSS', 'BaseDetector', 'SingleStageDetector', 'TwoStageDetector', 'RPN', |
|
'KnowledgeDistillationSingleStageDetector', 'FastRCNN', 'FasterRCNN', |
|
'MaskRCNN', 'CascadeRCNN', 'HybridTaskCascade', 'RetinaNet', 'FCOS', |
|
'GridRCNN', 'MaskScoringRCNN', 'RepPointsDetector', 'FOVEA', 'FSAF', |
|
'NASFCOS', 'PointRend', 'GFL', 'CornerNet', 'PAA', 'YOLOV3', 'YOLACT', |
|
'VFNet', 'DETR', 'TridentFasterRCNN', 'SparseRCNN', 'SCNet', 'SOLO', |
|
'DeformableDETR', 'AutoAssign', 'YOLOF', 'CenterNet', 'YOLOX', |
|
'TwoStagePanopticSegmentor', 'PanopticFPN' |
|
]
|
|
|