updates
This commit is contained in:
parent
578e7f9500
commit
dc1f0a0d4f
|
@ -866,7 +866,7 @@ def apply_classifier(x, model, img, im0):
|
||||||
|
|
||||||
def fitness(x):
|
def fitness(x):
|
||||||
# Returns fitness (for use with results.txt or evolve.txt)
|
# Returns fitness (for use with results.txt or evolve.txt)
|
||||||
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]
|
w = [0.01, 0.01, 0.78, 0.20] # 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)
|
return (x[:, :4] * w).sum(1)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue