From 340e0371f8b46283475923815e8f13dd338b5537 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 27 Nov 2019 22:36:01 -1000 Subject: [PATCH] updates --- utils/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index 805f5799..1ded9eae 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -511,8 +511,8 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5): if n == 1: det_max.append(dc) # No NMS required if only 1 prediction continue - elif n > 100: - dc = dc[:100] # limit to first 100 boxes: https://github.com/ultralytics/yolov3/issues/117 + elif n > 500: + dc = dc[:500] # limit to first 100 boxes: https://github.com/ultralytics/yolov3/issues/117 # Non-maximum suppression if nms_style == 'OR': # default