Comment fix (#3058)

Co-authored-by: kepler62f <>
2021-05-07 18:28:55 +08:00 · 2021-05-07 18:28:55 +08:00 · f2de1ad2aa
parent 3ef3a95cfa
commit f2de1ad2aa
1 changed files with 2 additions and 2 deletions
--- 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]: