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

mAP corrected to per-class

This commit is contained in:
Glenn Jocher 2018-09-10 15:23:39 +02:00
parent cd753d23f7
commit 873abaeef4
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test.py
View File

@ -118,6 +118,7 @@ for batch_i, (imgs, targets) in enumerate(dataloader):
mAP = AP.mean()
# Append image mAP to list
mAPs.append(mAP)
# Print image mAP and running mean mAP
print('+ Sample [%d/%d] AP: %.4f (%.4f)' % (len(mAPs), len(dataloader) * opt.batch_size, mAP, np.mean(mAPs)))