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

Fix Classify confusion matrix (#3241)

Browse source
This commit is contained in:
Glenn Jocher 2023-06-18 03:07:45 +02:00 • committed by GitHub
parent 9c44560ec4
commit 9d38d2b044
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/utils/metrics.py
View file

@ -204,7 +204,7 @@ class ConfusionMatrix:
""" """
preds, targets = torch.cat(preds)[:, 0], torch.cat(targets) preds, targets = torch.cat(preds)[:, 0], torch.cat(targets)
for p, t in zip(preds.cpu().numpy(), targets.cpu().numpy()): for p, t in zip(preds.cpu().numpy(), targets.cpu().numpy()):
self.matrix[t][p] += 1 self.matrix[p][t] += 1
def process_batch(self, detections, labels): def process_batch(self, detections, labels):
""" """