This commit is contained in:
Glenn Jocher
2019-07-25 13:37:01 +02:00
parent a834377122
commit 666d772fa7
4 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -580,7 +580,7 @@ def kmeans_targets(path='./data/coco_64img.txt', n=9, img_size=320): # from uti
k = cluster.vq.kmeans(wh, n)[0]
k = k[np.argsort(k.prod(1))]
for x in k.ravel():
print('%.1f, ' % x, end='')
print('%.1f, ' % x, end='') # drop-in replacement for *.cfg anchors
# Plotting functions ---------------------------------------------------------------------------------------------------