Fix bug when training images without gt (#786)

open-mmlab · Mar 27, 2023 · 17f8abc · 17f8abc
1 parent 45f2036
commit 17f8abc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmrotate/structures/bbox/bbox_overlaps.py b/mmrotate/structures/bbox/bbox_overlaps.py
@@ -82,7 +82,8 @@ def fake_rbbox_overlaps(bboxes1: RotatedBoxes,
         assert rows == cols
 
     if rows * cols == 0:
-        return bboxes1.new(rows, 1) if is_aligned else bboxes1.new(rows, cols)
+        return bboxes1.tensor.new(
+            rows, 1) if is_aligned else bboxes1.tensor.new(rows, cols)
 
     # convert rbb to minimum circumscribed hbb in <cx, cy, w, h, t> format.
     fake_rbboxes1 = bboxes1.convert_to('hbox').convert_to('rbox')