Refactor structure of mmrotate (#558)

* update * fix lint * update * update * fix ut * Update test_angle_coder.py * fix * Update __init__.py * fix * fix ut bugs
open-mmlab · Oct 21, 2022 · 288481f · 288481f
1 parent 9543a4e
commit 288481f
Show file tree

Hide file tree

Showing 134 changed files with 899 additions and 6,179 deletions.
diff --git a/configs/csl/rotated_retinanet_obb_csl_gaussian_r50_fpn_fp16_1x_dota_le90.py b/configs/csl/rotated_retinanet_obb_csl_gaussian_r50_fpn_fp16_1x_dota_le90.py
@@ -12,7 +12,4 @@
             window='gaussian',
             radius=3),
         loss_angle=dict(
-            type='mmdet.SmoothFocalLoss',
-            gamma=2.0,
-            alpha=0.25,
-            loss_weight=0.8)))
+            type='SmoothFocalLoss', gamma=2.0, alpha=0.25, loss_weight=0.8)))
diff --git a/configs/rotated_atss/rotated_atss_r50_fpn_1x_dota_le90.py b/configs/rotated_atss/rotated_atss_r50_fpn_1x_dota_le90.py
@@ -12,7 +12,7 @@
         std=[58.395, 57.12, 57.375],
         bgr_to_rgb=True,
         pad_size_divisor=32,
-        boxlist2tensor=False),
+        boxtype2tensor=False),
     backbone=dict(
         type='mmdet.ResNet',
         depth=50,

diff --git a/mmrotate/apis/__init__.py b/mmrotate/apis/__init__.py
@@ -1,4 +1,6 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+from patch import *  # noqa: F401, F403
+
 from .inference import inference_detector_by_patches
 
 __all__ = ['inference_detector_by_patches']
diff --git a/mmrotate/apis/inference.py b/mmrotate/apis/inference.py
@@ -7,7 +7,7 @@
 from mmdet.datasets import replace_ImageToTensor
 from mmdet.datasets.pipelines import Compose
 
-from mmrotate.core import get_multiscale_patch, merge_results, slide_window
+from mmrotate.apis import get_multiscale_patch, merge_results, slide_window
 
 
 def inference_detector_by_patches(model,

diff --git a/mmrotate/core/patch/__init__.py → mmrotate/apis/patch/__init__.py b/mmrotate/core/patch/__init__.py → mmrotate/apis/patch/__init__.py
diff --git a/mmrotate/core/patch/merge_results.py → mmrotate/apis/patch/merge_results.py b/mmrotate/core/patch/merge_results.py → mmrotate/apis/patch/merge_results.py
diff --git a/mmrotate/core/patch/split.py → mmrotate/apis/patch/split.py b/mmrotate/core/patch/split.py → mmrotate/apis/patch/split.py
diff --git a/mmrotate/core/__init__.py b/mmrotate/core/__init__.py
diff --git a/mmrotate/core/anchor/utils.py b/mmrotate/core/anchor/utils.py
diff --git a/mmrotate/core/bbox/__init__.py b/mmrotate/core/bbox/__init__.py
diff --git a/mmrotate/core/bbox/iou_calculators/__init__.py b/mmrotate/core/bbox/iou_calculators/__init__.py
diff --git a/mmrotate/core/bbox/samplers/__init__.py b/mmrotate/core/bbox/samplers/__init__.py
diff --git a/mmrotate/core/bbox/samplers/rotate_random_sampler.py b/mmrotate/core/bbox/samplers/rotate_random_sampler.py