From 310642c260c23db46dc124bebdd2d511f568c103 Mon Sep 17 00:00:00 2001
From: Dhananjai Sharma <31407075+dhananjaisharma10@users.noreply.github.com>
Date: Tue, 9 Jul 2019 12:42:53 -0400
Subject: [PATCH] Fix for issue #379

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

diff --git a/test.py b/test.py
index 85181a212..70a0c5f67 100644
--- a/test.py
+++ b/test.py
@@ -61,7 +61,7 @@ def test_net(save_folder, net, cuda, testset, transform, thresh):
         pred_num = 0
         for i in range(detections.size(1)):
             j = 0
-            while detections[0, i, j, 0] >= 0.6:
+            while detections[0, i, j, 0] >= thresh:
                 if pred_num == 0:
                     with open(filename, mode='a') as f:
                         f.write('PREDICTIONS: '+'\n')