diff --git a/utils/utils.py b/utils/utils.py index c396be03..5a1a4912 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -459,7 +459,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=False, method='vision'): """ Removes detections with lower object confidence score than 'conf_thres' Non-Maximum Suppression to further filter detections.