From e7019798623e683cf9e18374c4a0162574047617 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 22 Nov 2019 13:03:29 -1000 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 10502174..a9f77a1c 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -688,7 +688,7 @@ def kmeans_targets(path='../coco/trainvalno5k.txt', n=9, img_size=416): # from wh = np.concatenate(dataset.labels, 0)[:, 3:5] # wh from cxywh # Kmeans calculation - k = cluster.vq.kmeans(wh, n)[0] + k, dist = cluster.vq.kmeans(wh, n) # points, mean distance k = k[np.argsort(k.prod(1))] # sort small to large # Measure IoUs