randomaffine bbox coordinate recorrection (open-mmlab#6293)

jozhang97 · Oct 25, 2021 · 71a1cf2 · 71a1cf2
1 parent 9874180
commit 71a1cf2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet/datasets/pipelines/transforms.py b/mmdet/datasets/pipelines/transforms.py
@@ -2536,7 +2536,7 @@ def __call__(self, results):
             num_bboxes = len(bboxes)
             if num_bboxes:
                 # homogeneous coordinates
-                xs = bboxes[:, [0, 2, 2, 0]].reshape(num_bboxes * 4)
+                xs = bboxes[:, [0, 0, 2, 2]].reshape(num_bboxes * 4)
                 ys = bboxes[:, [1, 3, 3, 1]].reshape(num_bboxes * 4)
                 ones = np.ones_like(xs)
                 points = np.vstack([xs, ys, ones])