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

…atrix (#2883)
ultralytics · Apr 21, 2021 · d48a34d · d48a34d
1 parent 5f7d39f
commit d48a34d
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