Comment fix (ultralytics#3058)

Co-authored-by: kepler62f <> (cherry picked from commit f2de1ad)
Lechtr · May 24, 2021 · 6cfaf20 · 6cfaf20
1 parent 3466f04
commit 6cfaf20
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test.py b/test.py
@@ -188,8 +188,8 @@ def test(data,
 
                 # Per target class
                 for cls in torch.unique(tcls_tensor):
-                    ti = (cls == tcls_tensor).nonzero(as_tuple=False).view(-1)  # prediction indices
-                    pi = (cls == pred[:, 5]).nonzero(as_tuple=False).view(-1)  # target indices
+                    ti = (cls == tcls_tensor).nonzero(as_tuple=False).view(-1)  # target indices
+                    pi = (cls == pred[:, 5]).nonzero(as_tuple=False).view(-1)  # prediction indices
 
                     # Search for detections
                     if pi.shape[0]: