From eac2c010c461a2dffef211435dc27c8e6b34b321 Mon Sep 17 00:00:00 2001 From: Marc Date: Wed, 18 Dec 2019 16:13:20 -0500 Subject: [PATCH] return kmeans targets (#722) return kmeans targets --- utils/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/utils.py b/utils/utils.py index d7b27171..6c6901fc 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -751,6 +751,7 @@ def kmeans_targets(path='../coco/trainvalno5k.txt', n=9, img_size=416): # from # Plot # plt.hist(biou.numpy().ravel(), 100) + return k def print_mutation(hyp, results, bucket=''):