This commit is contained in:
Glenn Jocher
2020-01-10 12:49:22 -08:00
parent 3505b57421
commit c8a67adecc
2 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -865,7 +865,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]
w = [0.0, 0.0, 0.8, 0.2] # 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)