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

updates

This commit is contained in:
Glenn Jocher 2019-05-13 14:41:17 +02:00
parent 584e0a3be8
commit 3a2da49f82
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -176,7 +176,7 @@ def test(
map = cocoEval.stats[1] # update mAP to pycocotools mAP map = cocoEval.stats[1] # update mAP to pycocotools mAP
# Return results # Return results
maps = np.zeros(nc) maps = np.zeros(nc) + map
for i, c in enumerate(ap_class): for i, c in enumerate(ap_class):
maps[c] = ap[i] maps[c] = ap[i]
return (mp, mr, map, mf1, loss / len(dataloader)), maps return (mp, mr, map, mf1, loss / len(dataloader)), maps