From 4aae1fc5a58989868e0fd1ab54c21660d21ac0cc Mon Sep 17 00:00:00 2001 From: Qingyun <79644233+Li-Qingyun@users.noreply.github.com> Date: Tue, 20 Jun 2023 13:51:27 +0800 Subject: [PATCH] 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. --- mmrotate/models/dense_heads/h2rbox_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmrotate/models/dense_heads/h2rbox_head.py b/mmrotate/models/dense_heads/h2rbox_head.py index c66f6b29b..b904f1624 100644 --- 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