fix box error in h2rbox_head.py (#881)

There seems to be a lack of indentation, resulting in an assignment bug when square classes are not processed.
open-mmlab · Jun 20, 2023 · 4aae1fc · 4aae1fc
1 parent a860946
commit 4aae1fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmrotate/models/dense_heads/h2rbox_head.py b/mmrotate/models/dense_heads/h2rbox_head.py
@@ -703,5 +703,5 @@ def _predict_by_feat_single(self,
                 inds = results.labels == id
                 bboxes[inds, :] = hbox2rbox(rbox2hbox(bboxes[inds, :]))
 
-        results.bboxes = RotatedBoxes(bboxes)
+            results.bboxes = RotatedBoxes(bboxes)
         return results