fix empty

pyronear · Jan 31, 2025 · a753721 · a753721
1 parent 4416fa2
commit a753721
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyroengine/engine.py b/pyroengine/engine.py
@@ -242,6 +242,7 @@ def _update_states(self, frame: Image.Image, preds: np.ndarray, cam_key: str) ->
 
                     if len(output_predictions) == 0:
                         missing_bbox = combine_predictions
+                        missing_bbox[:, -1] = 0
 
                     else:
                         # Add missing bboxes