Fix AP calculation bug ultralytics#8464 (ultralytics#8484)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
ctjanuhowski · Sep 8, 2022 · 5f3a442 · 5f3a442
1 parent 2276307
commit 5f3a442
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/val.py b/val.py
@@ -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