Fix AP calculation bug #8464 (#8484)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
UnglvKitDeandGlenn Jocher authored and GitHub committed 2022-07-07 20:42:09 +02:00
1 parent dd28df98c2
commit 39d7a93619
1 file changed
+1 -1
+1 -1
View File
@@ -227,7 +227,7 @@ def run(
if npr == 0:
if nl:
stats.append((correct, *torch.zeros((3, 0), device=device)))
stats.append((correct, *torch.zeros((2, 0), device=device), labels[:, 0]))
continue
# Predictions