Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2020-01-08 16:39:59 -08:00
parent 3e5b007e3a
commit fd8cd377c3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/utils.py
View File

@ -864,7 +864,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]
return (x[:, :4] * np.array([w])).sum(1)
return (x[:, :4] * w).sum(1)
# Plotting functions ---------------------------------------------------------------------------------------------------