[Fix] compute distance to origin after sort (#1500)

2021-12-22 10:13:22 +08:00 · 2021-12-22 10:13:22 +08:00 · a0b87245f3
parent e4b5348ebf
commit a0b87245f3
1 changed files with 4 additions and 0 deletions
--- a/mmcv/ops/csrc/common/box_iou_rotated_utils.hpp
+++ b/mmcv/ops/csrc/common/box_iou_rotated_utils.hpp
@ -220,6 +220,10 @@ HOST_DEVICE_INLINE int convex_hull_graham(const Point<T> (&p)[24],
                return temp > 0;
              }
            });
+  // compute distance to origin after sort, since the points are now different.
+  for (int i = 0; i < num_in; i++) {
+    dist[i] = dot_2d<T>(q[i], q[i]);
+  }
 #endif

  // Step 4: