Merge branch 'develop' into ros1-update

opendr-eu · Aug 18, 2022 · 33dbc04 · 33dbc04
2 parents 8ec195a + f106e30
commit 33dbc04
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/opendr/perception/object_detection_2d/utils/eval_utils.py b/src/opendr/perception/object_detection_2d/utils/eval_utils.py
@@ -267,7 +267,8 @@ def update(self, det_boxes, det_labels, det_scores, gt_boxes, gt_labels, gt_diff
                 if score < self.score_threshold:
                     continue
                 img_dets.append(np.asarray([image['id'], box[0], box[1], box[2] - box[0], box[3] - box[1], score, cls]))
-            self.detections.append(np.asarray(img_dets))
+            if img_dets:
+                self.detections.append(np.asarray(img_dets))
 
             for box_idx, box in enumerate(gt_boxes[idx, :, :]):
                 cls = gt_labels[idx, box_idx]