From 28c4866b68c8c90293003c3eafdab7ac5a8ee3d8 Mon Sep 17 00:00:00 2001
From: kepler62f <>
Date: Fri, 7 May 2021 10:27:14 +0800
Subject: [PATCH] Fixed #3042

---
 test.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test.py b/test.py
index 0093b8c09f54..f8936d3b4f9d 100644
--- 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]: