From a96285870de3a06f12842af33bd2b9e041375c9d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 22 Dec 2019 13:04:44 -0800 Subject: [PATCH] updates --- utils/utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index 5c7dd82b..e3f1f7e5 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -466,8 +466,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5, multi_cls=Tru Returns detections with shape: (x1, y1, x2, y2, object_conf, conf, class) """ - # NMS method https://github.com/ultralytics/yolov3/issues/679 'or', 'and', 'merge', 'vision', 'vision_batch' - # method = 'merge' if conf_thres <= 0.01 else 'vision' # MERGE is highest mAP, VISION is fastest + # NMS methods https://github.com/ultralytics/yolov3/issues/679 'or', 'and', 'merge', 'vision', 'vision_batch' # Box constraints min_wh, max_wh = 2, 10000 # (pixels) minimum and maximium box width and height