Skip to content

Commit cd235e7

Browse files
authored
fix bugs
1 parent 8e47b39 commit cd235e7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

confusion_matrix.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,12 +77,12 @@ def process_batch(self, detections, labels):
7777
self.matrix[(gt_class), detection_class] += 1
7878
else:
7979
gt_class = gt_classes[i]
80-
self.matrix[(gt_class), self.num_classes] += 1
80+
self.matrix[self.num_classes, (gt_class)] += 1
8181

8282
for i, detection in enumerate(detections):
8383
if all_matches.shape[0] and all_matches[all_matches[:, 1] == i].shape[0] == 0:
8484
detection_class = detection_classes[i]
85-
self.matrix[self.num_classes ,detection_class] += 1
85+
self.matrix[detection_class, self.num_classes] += 1
8686

8787
def return_matrix(self):
8888
return self.matrix

0 commit comments

Comments
 (0)