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

Merge NMS update

This commit is contained in:
Glenn Jocher 2020-03-26 12:48:00 -07:00
parent 171b4129b5
commit a322fc5d4b
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
utils/utils.py
View File

@ -564,6 +564,7 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, multi_label=T
i = iou.max(0)[0] < iou_thres i = iou.max(0)[0] < iou_thres
output[xi] = x[i] output[xi] = x[i]
return output
def get_yolo_layers(model): def get_yolo_layers(model):