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

updates

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -197,7 +197,7 @@ def test(cfg,
cocoEval.evaluate() cocoEval.evaluate()
cocoEval.accumulate() cocoEval.accumulate()
cocoEval.summarize() cocoEval.summarize()
map = cocoEval.stats[1] # update mAP to pycocotools mAP mf1, map = cocoEval.stats[:2] # update to pycocotools results (mAP@0.5:0.95, mAP@0.5)
# Return results # Return results
maps = np.zeros(nc) + map maps = np.zeros(nc) + map

2
utils/utils.py
View File

@ -810,7 +810,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)
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 --------------------------------------------------------------------------------------------------- # Plotting functions ---------------------------------------------------------------------------------------------------