Skip to content

Commit 2846fd8

Browse files
authored
Merge pull request #395 from iocalangiu/cocomap-bug-no-gts
corrected bug for not counts gts
2 parents 5ac6f37 + 46ef951 commit 2846fd8

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

perceptionmetrics/utils/detection_metrics.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,10 @@ def update(self, gt_boxes, gt_labels, pred_boxes, pred_labels, pred_scores):
6666
(gt_boxes, gt_labels, pred_boxes, pred_labels, pred_scores)
6767
)
6868

69+
# Update ground truth counts
70+
for g_label in gt_labels:
71+
self.gt_counts[int(g_label)] += 1
72+
6973
# Handle empty inputs
7074
if len(gt_boxes) == 0 and len(pred_boxes) == 0:
7175
return # Nothing to process
@@ -89,9 +93,6 @@ def update(self, gt_boxes, gt_labels, pred_boxes, pred_labels, pred_scores):
8993
for label in matches:
9094
self.results[label].extend(matches[label])
9195

92-
# Update ground truth counts
93-
for g_label in gt_labels:
94-
self.gt_counts[int(g_label)] += 1
9596

9697
def _match_predictions(
9798
self,

0 commit comments

Comments
 (0)