From af23270482fd807dcf06e329b29c3d722b71d2a9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 6 Jan 2020 13:57:20 -0800 Subject: [PATCH] updates --- utils/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index 5bc84fff..2171d15b 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -773,7 +773,8 @@ def kmean_anchors(path='../coco/train2017.txt', n=9, img_size=(320, 640)): wh *= np.random.uniform(img_size[0], img_size[1], size=(wh.shape[0], 1)) # normalized to pixels (multi-scale) # Darknet yolov3.cfg anchors - if n == 9: + use_darknet = False + if use_darknet: k = np.array([[10, 13], [16, 30], [33, 23], [30, 61], [62, 45], [59, 119], [116, 90], [156, 198], [373, 326]]) k = print_results(thr, wh, k) else: