return kmeans targets (#722)

return kmeans targets
2019-12-18 16:13:20 -05:00 · 2019-12-18 16:13:20 -05:00 · eac2c010c4
parent ad73ce4334
commit eac2c010c4
1 changed files with 1 additions and 0 deletions
--- 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=''):