This commit is contained in:
Glenn Jocher 2020-01-06 14:16:23 -08:00
parent fd0769c476
commit 3b5ca2ea90
1 changed files with 1 additions and 2 deletions

View File

@ -776,7 +776,6 @@ def kmean_anchors(path='../coco/train2017.txt', n=9, img_size=(320, 640)):
use_darknet = False use_darknet = False
if use_darknet: 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 = 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: else:
# Kmeans calculation # Kmeans calculation
from scipy.cluster.vq import kmeans from scipy.cluster.vq import kmeans