Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-05-03 00:26:26 +02:00
parent b901441e76
commit 3c55d63a9d
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/gcp.sh
View File

@ -55,7 +55,7 @@ gsutil cp results.png gs://ultralytics
sudo shutdown sudo shutdown
# Reproduce mAP # 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 416
python3 test.py --save-json --img-size 320 python3 test.py --save-json --img-size 320
sudo shutdown sudo shutdown

2
utils/utils.py
View File

@ -431,7 +431,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5):
dc = dc[i == 0] dc = dc[i == 0]
elif nms_style == 'SOFT': # soft-NMS https://arxiv.org/abs/1704.04503 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): while len(dc):
if len(dc) == 1: if len(dc) == 1:
det_max.append(dc) det_max.append(dc)