Update kmeans_for_anchors.py

bubbliiiing · Jan 15, 2022 · 02b0e1f · 02b0e1f
1 parent ddf9d16
commit 02b0e1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kmeans_for_anchors.py b/kmeans_for_anchors.py
@@ -150,7 +150,7 @@ def load_data(path):
 
     cluster = cluster[np.argsort(cluster[:, 0] * cluster[:, 1])]
     print('avg_ratio:{:.2f}'.format(avg_iou(data, cluster)))
-    print(data)
+    print(cluster)
 
     f = open("yolo_anchors.txt", 'w')
     row = np.shape(cluster)[0]