fix bug in the ConfusionMatrix class (#18409)

2024-12-26 21:23:03 +08:00 · 2024-12-26 21:23:03 +08:00 · edf8ba0f17
commit edf8ba0f17
parent 7afec0d89e
1 changed files with 3 additions and 4 deletions
--- a/ultralytics/utils/metrics.py
+++ b/ultralytics/utils/metrics.py
@ -372,7 +372,6 @@ class ConfusionMatrix:
            else:
                self.matrix[self.nc, gc] += 1  # true background

-        if n:
        for i, dc in enumerate(detection_classes):
            if not any(m1 == i):
                self.matrix[dc, self.nc] += 1  # predicted background