From 8841c4980cb88a430b715658fdef20b410140b76 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 2 Jan 2020 10:03:22 -0800 Subject: [PATCH] updates --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index 684fbf48..5d97f33f 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -770,7 +770,7 @@ def kmean_anchors(path='data/coco64.txt', n=12, img_size=(320, 640)): iou = wh_iou(torch.Tensor(wh), torch.Tensor(k)) min_iou, max_iou = iou.min(1)[0], iou.max(1)[0] for x in [0.10, 0.15, 0.20, 0.25, 0.30, 0.35]: # iou thresholds - print('%.2f iou_thr: %.3f best possible recall, %.3f anchors > thr' % + print('%.2f iou_thr: %.3f best possible recall, %.1f anchors > thr' % (x, (max_iou > x).float().mean(), (iou > x).float().mean() * n)) # BPR (best possible recall) # Print