Add files via upload

add r3det and s2anet crop

mmrotate/models/detectors/__init__.py

@@ -1,28 +1,31 @@
-# Copyright (c) OpenMMLab. All rights reserved.
-from .base import RotatedBaseDetector
-from .gliding_vertex import GlidingVertex
-from .oriented_rcnn import OrientedRCNN
-from .r3det import R3Det
-from .redet import ReDet
-from .roi_transformer import RoITransformer
-from .rotate_faster_rcnn import RotatedFasterRCNN
-from .rotated_fcos import RotatedFCOS
-from .rotated_reppoints import RotatedRepPoints
-from .rotated_retinanet import RotatedRetinaNet
-from .s2anet import S2ANet
-from .single_stage import RotatedSingleStageDetector
-from .single_stage_crop import RotatedSingleStageDetectorCrop
-from .two_stage import RotatedTwoStageDetector
-from .two_stage_crop import RotatedTwoStageDetectorCrop
-from .rotated_detr import RotatedDETR
-from .rotated_deformable_detr import RotatedDeformableDETR
-from .ars_detr import ARSDETR
-from .h2rbox import H2RBox
-
-__all__ = [
-    'RotatedRetinaNet', 'RotatedFasterRCNN', 'OrientedRCNN', 'RoITransformer',
-    'GlidingVertex', 'ReDet', 'R3Det', 'S2ANet', 'RotatedRepPoints',
-    'RotatedBaseDetector', 'RotatedSingleStageDetectorCrop', 'RotatedTwoStageDetector',
-    'RotatedSingleStageDetector', 'RotatedFCOS', 'RotatedTwoStageDetectorCrop',
-    'RotatedDETR', 'RotatedDeformableDETR', 'ARSDETR', 'H2RBox'
-]
+# Copyright (c) OpenMMLab. All rights reserved.
+from .base import RotatedBaseDetector
+from .gliding_vertex import GlidingVertex
+from .oriented_rcnn import OrientedRCNN
+from .r3det import R3Det
+from .redet import ReDet
+from .roi_transformer import RoITransformer
+from .rotate_faster_rcnn import RotatedFasterRCNN
+from .rotated_fcos import RotatedFCOS
+from .rotated_reppoints import RotatedRepPoints
+from .rotated_retinanet import RotatedRetinaNet
+from .s2anet import S2ANet
+from .single_stage import RotatedSingleStageDetector
+from .single_stage_crop import RotatedSingleStageDetectorCrop
+from .two_stage import RotatedTwoStageDetector
+from .two_stage_crop import RotatedTwoStageDetectorCrop
+from .rotated_detr import RotatedDETR
+from .rotated_deformable_detr import RotatedDeformableDETR
+from .ars_detr import ARSDETR
+from .h2rbox import H2RBox
+
+from .r3det_crop import R3DetCrop
+from .s2anet_crop import S2ANetCrop
+
+__all__ = [
+    'RotatedRetinaNet', 'RotatedFasterRCNN', 'OrientedRCNN', 'RoITransformer',
+    'GlidingVertex', 'ReDet', 'R3Det', 'S2ANet', 'RotatedRepPoints',
+    'RotatedBaseDetector', 'RotatedSingleStageDetectorCrop', 'RotatedTwoStageDetector',
+    'RotatedSingleStageDetector', 'RotatedFCOS', 'RotatedTwoStageDetectorCrop',
+    'RotatedDETR', 'RotatedDeformableDETR', 'ARSDETR', 'H2RBox','R3DetCrop','S2ANetCrop'
+]

mmrotate/models/detectors/r3det_crop.py

@@ -177,7 +177,7 @@ def relocate(idx, local_bboxes, patch_meta):
 
 
 @ROTATED_DETECTORS.register_module()
-class R3Det(RotatedBaseDetector):
+class R3DetCrop(RotatedBaseDetector):
     """Rotated Refinement RetinaNet."""
 
     def __init__(self,
@@ -191,7 +191,7 @@ def __init__(self,
                  test_cfg=None,
                  pretrained=None,
                  init_cfg=None):
-        super(R3Det, self).__init__(init_cfg)
+        super(R3DetCrop, self).__init__(init_cfg)
         if pretrained:
             warnings.warn('DeprecationWarning: pretrained is deprecated, '
                           'please use "init_cfg" instead')