From bf42c31d9ee10090a61ebfa2192098d6d6383695 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 6 Jan 2020 14:25:11 -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 059fba84..60efd25a 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -761,7 +761,7 @@ def kmean_anchors(path='../coco/train2017.txt', n=9, img_size=(320, 640)): def fitness(thr, wh, k): # mutation fitness iou = wh_iou(wh, torch.Tensor(k)).max(1)[0] # max iou bpr = (iou > thr).float().mean() # best possible recall - return iou.mean() * 0.80 + bpr * 0.20 # weighted combination + return iou.mean() * bpr # product # Get label wh wh = []