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-30 13:46:21 -08:00
parent cf92235b8d
commit 7b6bd39c9e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
utils/utils.py
View File

@ -762,7 +762,8 @@ def kmean_anchors(path='data/coco64.txt', n=9, img_size=(288, 640)): # from uti
# Measure IoUs # Measure IoUs
iou = wh_iou(torch.Tensor(wh), torch.Tensor(k)) iou = wh_iou(torch.Tensor(wh), torch.Tensor(k))
biou = iou.max(0)[0] # closest anchor IoU biou = iou.max(1)[0] # closest anchor IoU
print(biou.shape)
print('Best Possible Recall (BPR): %.3f' % (biou > 0.225).float().mean()) # BPR (best possible recall) print('Best Possible Recall (BPR): %.3f' % (biou > 0.225).float().mean()) # BPR (best possible recall)
# Print # Print