fix: nms tests

Smirkey · Jan 3, 2024 · fe2b479 · fe2b479
1 parent 13e23f1
commit fe2b479
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/powerboxesrs/src/nms.rs b/powerboxesrs/src/nms.rs
@@ -38,8 +38,8 @@ pub fn nms<N>(
 where
     N: Num + PartialOrd + ToPrimitive + Copy,
 {
-    let mut above_score_threshold: Vec<usize> = (1..scores.len()).collect();
-    if score_threshold < utils::EPS {
+    let mut above_score_threshold: Vec<usize> = (0..scores.len()).collect();
+    if score_threshold > utils::EPS {
         // filter out boxes lower than score threshold
         above_score_threshold = scores
             .iter()
@@ -161,8 +161,8 @@ pub fn rtree_nms<N>(
 where
     N: RTreeNum + ToPrimitive + Send + Sync,
 {
-    let mut above_score_threshold: Vec<usize> = (1..scores.len()).collect();
-    if score_threshold < utils::EPS {
+    let mut above_score_threshold: Vec<usize> = (0..scores.len()).collect();
+    if score_threshold > utils::EPS {
         // filter out boxes lower than score threshold
         above_score_threshold = scores
             .iter()