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

updates

This commit is contained in:
Glenn Jocher 2019-04-09 16:28:14 +02:00
parent 2ca4c9aaec
commit 7709e8aa72
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
utils/utils.py
View File

@ -319,13 +319,14 @@ def build_targets(model, targets):
# Class
tcls.append(c)
assert c.max() <= layer.nC, 'Target classes exceed model classes'
if c.shape[0]:
assert c.max() <= layer.nC, 'Target classes exceed model classes'
return txy, twh, tcls, indices
# @profile
def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.4):
def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5):
"""
Removes detections with lower object confidence score than 'conf_thres'
Non-Maximum Suppression to further filter detections.