From 4aa60ea4992208ae59aa201ebaaaf2d21f318c57 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 27 Sep 2019 23:40:14 +0200 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 c55ba4bf..8b820a97 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -451,7 +451,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5): min_wh = 2 # (pixels) minimum box width and height output = [None] * len(prediction) - for image_i, pred in enumerate(prediction.float()): + for image_i, pred in enumerate(prediction): # Experiment: Prior class size rejection # x, y, w, h = pred[:, 0], pred[:, 1], pred[:, 2], pred[:, 3] # a = w * h # area