From 71a1cf2b46f2f7b2dc124774f747ee3754c72d78 Mon Sep 17 00:00:00 2001 From: DapangpangX <40031552+DapangpangX@users.noreply.github.com> Date: Mon, 25 Oct 2021 11:07:12 +0800 Subject: [PATCH] randomaffine bbox coordinate recorrection (#6293) --- mmdet/datasets/pipelines/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmdet/datasets/pipelines/transforms.py b/mmdet/datasets/pipelines/transforms.py index 7f99b97aa67..cb5945d2832 100644 --- 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])