From a322fc5d4beb0cffc1b3582945bf44ef32a66baa Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 26 Mar 2020 12:48:00 -0700 Subject: [PATCH] Merge NMS update --- utils/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/utils.py b/utils/utils.py index b6b15b84..820bd936 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -564,6 +564,7 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, multi_label=T i = iou.max(0)[0] < iou_thres output[xi] = x[i] + return output def get_yolo_layers(model):