Merge pull request #62 from tralfamadude/master

fix n_max_boxes off by one
dhlab-epfl · Dec 17, 2020 · cca94e9 · cca94e9
2 parents 823b6b4 + 037f222
commit cca94e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dh_segment/post_processing/boxes_detection.py b/dh_segment/post_processing/boxes_detection.py
@@ -105,4 +105,4 @@ def validate_box(box: np.array) -> (np.array, float):
         else:
             return None
     else:
-        return [fb[0] for i, fb in enumerate(found_boxes) if i <= n_max_boxes]
+        return [fb[0] for i, fb in enumerate(found_boxes) if i < n_max_boxes]