Fix confusion matrix update when no predictions are made (#8748)

* Fix confusion matrix update when no predictions are made

* Update metrics.py

* Simply confusion matrix changes

* Simply confusion matrix fix

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
jbutle55andGlenn Jocher authored and GitHub committed 2022-07-29 14:06:23 +02:00
1 parent 587a3a37c5
commit 567397d67a
2 files changed
+8

No files matched your search

+2
View File
@@ -228,6 +228,8 @@ def run(
if npr == 0:
if nl:
stats.append((correct, *torch.zeros((2, 0), device=device), labels[:, 0]))
if plots:
confusion_matrix.process_batch(detections=None, labels=labels[:, 0])
continue
# Predictions