From fd8cd377c30bdd9651fbb3e28627fdc65e20a0c1 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 8 Jan 2020 16:39:59 -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 502688b4..c93cecdf 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -864,7 +864,7 @@ def apply_classifier(x, model, img, im0): def fitness(x): # Returns fitness (for use with results.txt or evolve.txt) w = [0.1, 0.1, 0.6, 0.2] # weights for [P, R, mAP, F1]@0.5 or [P, R, mAP@0.5:0.95, mAP@0.5] - return (x[:, :4] * np.array([w])).sum(1) + return (x[:, :4] * w).sum(1) # Plotting functions ---------------------------------------------------------------------------------------------------