fix empty

pyronear · Jan 31, 2025 · 10ff08d · 10ff08d
1 parent 9f5d987
commit 10ff08d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyroengine/engine.py b/pyroengine/engine.py
@@ -252,7 +252,7 @@ def _update_states(self, frame: Image.Image, preds: np.ndarray, cam_key: str) ->
                     output_predictions = output_predictions[:5, :]  # max 5 bbox
 
         # Add default bbox
-        if output_predictions.shape[0] == 0:
+        if len(output_predictions) == 0:
             output_predictions = np.zeros((1, 5))
             output_predictions[:, 2:4] += 0.0001