updates

2020-01-08 16:36:35 -08:00
parent 11ce877bdf
commit 3e5b007e3a
2 changed files with 4 additions and 3 deletions
@@ -863,7 +863,8 @@ def apply_classifier(x, model, img, im0):

 def fitness(x):
    # Returns fitness (for use with results.txt or evolve.txt)
-    return x[:, 2] * 0.3 + x[:, 3] * 0.7  # weighted combination of x=[p, r, mAP@0.5, F1 or mAP@0.5:0.95]
+    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)


 # Plotting functions ---------------------------------------------------------------------------------------------------