diff --git a/utils/utils.py b/utils/utils.py index 1fd88e9c..f78534da 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -872,7 +872,7 @@ def apply_classifier(x, model, img, im0): def fitness(x): # Returns fitness (for use with results.txt or evolve.txt) - w = [0.01, 0.01, 0.75, 0.23] # 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.0] # weights for [P, R, mAP, F1]@0.5 or [P, R, mAP@0.5:0.95, mAP@0.5] return (x[:, :4] * w).sum(1)