updates
This commit is contained in:
parent
7b6bd39c9e
commit
2cf31ab7bc
|
@ -763,7 +763,6 @@ 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(1)[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
|
||||||
|
|
Loading…
Reference in New Issue