This commit is contained in:
Glenn Jocher
2019-12-16 16:29:40 -08:00
parent d7b010c514
commit 8666413c47
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -810,7 +810,7 @@ def apply_classifier(x, model, img, im0):
def fitness(x):
# Returns fitness (for use with results.txt or evolve.txt)
return x[:, 2] * 0.8 + x[:, 3] * 0.2 # weighted mAP and F1 combination
return x[:, 2] * 0.5 + x[:, 3] * 0.5 # weighted combination of x=[p, r, mAP@0.5, F1 or mAP@0.5:0.95]
# Plotting functions ---------------------------------------------------------------------------------------------------