edit in comments (#1417)

Co-authored-by: Priteshkumar Bharatbhai Gohil <pgohil@assystemtechnologies.com>
2020-07-27 19:29:45 +02:00
parent 8de13f114d
commit e0a5a6b411
1 changed files with 2 additions and 2 deletions
@@ -145,8 +145,8 @@ def test(cfg,

                # Per target class
                for cls in torch.unique(tcls_tensor):
-                    ti = (cls == tcls_tensor).nonzero().view(-1)  # prediction indices
-                    pi = (cls == pred[:, 5]).nonzero().view(-1)  # target indices
+                    ti = (cls == tcls_tensor).nonzero().view(-1)  # target indices
+                    pi = (cls == pred[:, 5]).nonzero().view(-1)  # prediction indices

                    # Search for detections
                    if pi.shape[0]: