updates

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

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)