This commit is contained in:
Glenn Jocher 2019-12-24 13:59:20 -08:00
parent f7ac56db39
commit 2ee0d0c714
1 changed files with 1 additions and 1 deletions

View File

@ -497,7 +497,7 @@ def build_targets(model, targets):
return tcls, tbox, indices, av
def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5, multi_cls=True, method='vision'):
def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5, multi_cls=True, method='vision_batch'):
"""
Removes detections with lower object confidence score than 'conf_thres'
Non-Maximum Suppression to further filter detections.