From 121526aa9866ad752be222e9e1c619744c1d6bbf Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 30 Dec 2019 13:15:10 -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 c017bc81..ab27baf5 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -746,7 +746,7 @@ def kmeans_targets(path='data/coco64.txt', n=9, img_size=416): # from utils.uti l[:, [2, 4]] *= s[1] l[:, 1:] *= img_size / max(s) l = l.repeat(10, axis=0) # augment 10x - l *= np.random.uniform(0.5, 1.5, size=(l.shape[0], 1)) # multi-scale box + l *= np.random.uniform(288, 640, size=(l.shape[0], 1)) / img_size # multi-scale box wh = np.concatenate(dataset.labels, 0)[:, 3:5] # wh from cxywh # Kmeans calculation