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

updates

This commit is contained in:
Glenn Jocher 2019-11-27 22:36:01 -10:00
parent 9e9a6a1425
commit 340e0371f8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/utils.py
View File

@ -511,8 +511,8 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5):
if n == 1: if n == 1:
det_max.append(dc) # No NMS required if only 1 prediction det_max.append(dc) # No NMS required if only 1 prediction
continue continue
elif n > 100: elif n > 500:
dc = dc[:100] # limit to first 100 boxes: https://github.com/ultralytics/yolov3/issues/117 dc = dc[:500] # limit to first 100 boxes: https://github.com/ultralytics/yolov3/issues/117
# Non-maximum suppression # Non-maximum suppression
if nms_style == 'OR': # default if nms_style == 'OR': # default