From edf8ba0f172bab7461cc649ed8bf588f3c07d5bf Mon Sep 17 00:00:00 2001 From: yuzhj Date: Thu, 26 Dec 2024 21:23:03 +0800 Subject: [PATCH] fix bug in the ConfusionMatrix class (#18409) --- ultralytics/utils/metrics.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/ultralytics/utils/metrics.py b/ultralytics/utils/metrics.py index bb521f5c..392de427 100644 --- a/ultralytics/utils/metrics.py +++ b/ultralytics/utils/metrics.py @@ -372,10 +372,9 @@ 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 + for i, dc in enumerate(detection_classes): + if not any(m1 == i): + self.matrix[dc, self.nc] += 1 # predicted background def matrix(self): """Returns the confusion matrix."""