From be01fc357b67da6af1018dd309215d7f297b0eb5 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 4 Mar 2020 00:22:01 -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 1c1368b3..65ffbb1c 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -563,7 +563,7 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, multi_cls=Tru i = torchvision.ops.boxes.batched_nms(boxes, scores, c, iou_thres) elif method == 'fast_batch': # FastNMS from https://github.com/dbolya/yolact boxes += c.view(-1, 1) * max_wh - iou = box_iou(boxes, boxes).triu_(diagonal=1) # zero upper triangle iou matrix + iou = box_iou(boxes, boxes).triu_(diagonal=1) # upper triangular iou matrix i = iou.max(dim=0)[0] < iou_thres output[image_i] = pred[i]