From 3c55d63a9d6058404b04f22e2347ec31ddc5ecc5 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 3 May 2019 00:26:26 +0200
Subject: [PATCH] updates

---
 utils/gcp.sh   | 2 +-
 utils/utils.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/utils/gcp.sh b/utils/gcp.sh
index af9de6ee..ad90d3aa 100755
--- a/utils/gcp.sh
+++ b/utils/gcp.sh
@@ -55,7 +55,7 @@ gsutil cp results.png gs://ultralytics
 sudo shutdown
 
 # Reproduce mAP
-python3 test.py --save-json --img-size 608  --batch-size 16
+python3 test.py --save-json --img-size 608
 python3 test.py --save-json --img-size 416
 python3 test.py --save-json --img-size 320
 sudo shutdown
diff --git a/utils/utils.py b/utils/utils.py
index b0047217..d00b8aa4 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -431,7 +431,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5):
                     dc = dc[i == 0]
 
             elif nms_style == 'SOFT':  # soft-NMS https://arxiv.org/abs/1704.04503
-                sigma = nms_thres  # soft-nms sigma parameter
+                sigma = 0.5  # soft-nms sigma parameter
                 while len(dc):
                     if len(dc) == 1:
                         det_max.append(dc)