From 2ee0d0c71462ee801408f5e39acd3a2b3774ab71 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 24 Dec 2019 13:59:20 -0800 Subject: [PATCH] updates --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index 7c57d71d..312b8a95 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -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.