bug fix: switched rows and cols for correct detections in confusion m…

…atrix (ultralytics#2883) (cherry picked from commit d48a34d)
Lechtr · Apr 25, 2021 · c465df1 · c465df1
1 parent e0a9cb2
commit c465df1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/metrics.py b/utils/metrics.py
@@ -145,7 +145,7 @@ def process_batch(self, detections, labels):
         for i, gc in enumerate(gt_classes):
             j = m0 == i
             if n and sum(j) == 1:
-                self.matrix[gc, detection_classes[m1[j]]] += 1  # correct
+                self.matrix[detection_classes[m1[j]], gc] += 1  # correct
             else:
                 self.matrix[self.nc, gc] += 1  # background FP