diff --git a/utils/utils.py b/utils/utils.py index 61890c2e..85dd3658 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -878,7 +878,7 @@ def apply_classifier(x, model, img, im0): def fitness(x): # Returns fitness (for use with results.txt or evolve.txt) - w = [0.0, 0.01, 0.99, 0.0] # weights for [P, R, mAP, F1]@0.5 or [P, R, mAP@0.5:0.95, mAP@0.5] + w = [0.0, 0.01, 0.99, 0.00] # weights for [P, R, mAP, F1]@0.5 or [P, R, mAP@0.5, mAP@0.5:0.95] return (x[:, :4] * w).sum(1)