From 47eab968abdb3be99e638c25e4bad8590b62abb7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 26 Mar 2019 18:02:57 +0100 Subject: [PATCH] updates --- utils/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/utils.py b/utils/utils.py index 2233f88f..f33ac340 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -382,6 +382,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.4): # Sort the detections by maximum object confidence _, conf_sort_index = torch.sort(dc[:, 4] * dc[:, 5], descending=True) dc = dc[conf_sort_index] + dc = dc[:min(len(dc), 100)] # limit to first 100 boxes: https://github.com/ultralytics/yolov3/issues/117 # Non-maximum suppression det_max = []