fix(custom plugin): Resolved duplicate oriented bounding boxes in Rot…

…atedNMS plugin under FP16 precision (#71)
laugh12321 · Dec 18, 2024 · 6dba684 · 6dba684
1 parent bd3cbda
commit 6dba684
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 382 deletions.
diff --git a/plugin/efficientRotatedNMSPlugin/efficientRotatedNMSInference.cu b/plugin/efficientRotatedNMSPlugin/efficientRotatedNMSInference.cu
@@ -36,17 +36,7 @@ __device__ float IOU(EfficientRotatedNMSParameters param, RotatedBoxCorner<T> bo
     RotatedBoxCorner<T> b2 = box2;
     b1.reorder();
     b2.reorder();
-    float intersectArea = RotatedBoxCorner<T>::intersect_area(b1, b2);
-    if (intersectArea <= 0.f)
-    {
-        return 0.f;
-    }
-    float unionArea = b1.area() + b2.area() - intersectArea;
-    if (unionArea <= 0.f)
-    {
-        return 0.f;
-    }
-    return intersectArea / unionArea;
+    return RotatedBoxCorner<T>::probiou(b1, b2);
 }
 
 template <typename T, typename Tb>