prevent empty (#171)

pyronear · Aug 16, 2023 · 94a0947 · 94a0947
1 parent c04bff3
commit 94a0947
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyroengine/engine.py b/pyroengine/engine.py
@@ -245,7 +245,7 @@ def _update_states(self, frame: Image.Image, preds: np.array, cam_key: str) -> b
             conf = np.max(best_boxes_scores) / self.nb_consecutive_frames
 
             # if current predictions match with combine predictions send match else send combine predcition
-            ious = box_iou(combine_predictions[:, :4], preds[:, :4])[0]
+            ious = box_iou(combine_predictions[:, :4], preds[:, :4])
             if np.sum(ious) > 0:
                 output_predictions = preds
             else: