HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

fix bug in the ConfusionMatrix class (#18409)

Browse source
This commit is contained in:
yuzhj 2024-12-26 21:23:03 +08:00 • committed by GitHub
parent 7afec0d89e
commit edf8ba0f17
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
ultralytics/utils/metrics.py
View file

@ -372,10 +372,9 @@ class ConfusionMatrix:
else: else:
self.matrix[self.nc, gc] += 1 # true background self.matrix[self.nc, gc] += 1 # true background
if n: for i, dc in enumerate(detection_classes):
for i, dc in enumerate(detection_classes): if not any(m1 == i):
if not any(m1 == i): self.matrix[dc, self.nc] += 1 # predicted background
self.matrix[dc, self.nc] += 1 # predicted background
def matrix(self): def matrix(self):
"""Returns the confusion matrix.""" """Returns the confusion matrix."""