From 8b707c43c8a9ceefa51cf7c2d6c55e8de4dbbf49 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 18 Apr 2019 23:17:51 +0200
Subject: [PATCH] updates

---
 test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test.py b/test.py
index 3e058036..e78d6be1 100644
--- a/test.py
+++ b/test.py
@@ -125,7 +125,7 @@ def test(
                     iou, bi = bbox_iou(pbox, tbox).max(0)
 
                     # If iou > threshold and class is correct mark as correct
-                    if iou > iou_thres and bi not in detected and pcls == tcls[bi]:
+                    if iou > iou_thres and bi not in detected:  # and pcls == tcls[bi]:
                         correct[i] = 1
                         detected.append(bi)