fix: stop using utils::EPS in NMS

Smirkey · Apr 29, 2024 · 37f8ed4 · 37f8ed4
1 parent 7d2e5e0
commit 37f8ed4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/powerboxesrs/src/nms.rs b/powerboxesrs/src/nms.rs
@@ -108,7 +108,7 @@ where
             let intersection: N = area(x, y, xx, yy);
             let area2: N = area(b2x, b2y, b2xx, b2yy);
             let union: N = area1 + area2 - intersection;
-            let iou: f64 = intersection.to_f64().unwrap() / (union.to_f64().unwrap() + utils::EPS);
+            let iou: f64 = intersection.to_f64().unwrap() / union.to_f64().unwrap();
             if iou > iou_threshold {
                 suppress[j] = true;
             }
@@ -229,7 +229,7 @@ where
             let intersection: N = area(x, y, xx, yy);
             let area2: N = area(b2x, b2y, b2xx, b2yy);
             let union: N = area1 + area2 - intersection;
-            let iou: f64 = intersection.to_f64().unwrap() / (union.to_f64().unwrap() + utils::EPS);
+            let iou: f64 = intersection.to_f64().unwrap() / union.to_f64().unwrap();
             if iou > iou_threshold_f64 {
                 suppress[idx_j] = true;
             }