From 35eae3ace984595981136b899a76563a638f2a5b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 4 Mar 2020 09:53:02 -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 65ffbb1c..2f38ba73 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -523,7 +523,7 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, multi_cls=Tru pred = pred[pred[:, 4] > conf_thres] # Apply width-height constraint - pred = pred[(pred[:, 2:4] > min_wh).all(1) & (pred[:, 2:4] < max_wh).all(1)] + pred = pred[((pred[:, 2:4] > min_wh) & (pred[:, 2:4] < max_wh)).all(1)] # Compute conf pred[..., 5:] *= pred[..., 4:5] # conf = obj_conf * cls_conf