xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-10-31 14:35:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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
Show Stats 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):
""" """