From e0a5a6b411cca45f0d64aa932abffbf3c99b92b3 Mon Sep 17 00:00:00 2001 From: priteshgohil <43172056+priteshgohil@users.noreply.github.com> Date: Mon, 27 Jul 2020 19:29:45 +0200 Subject: [PATCH] edit in comments (#1417) Co-authored-by: Priteshkumar Bharatbhai Gohil --- test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test.py b/test.py index 3e739d8e..4d68438e 100644 --- a/test.py +++ b/test.py @@ -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]: