Skip to content

Commit 8655d04

Browse files
authored
Update kmeans_for_anchors.py
1 parent bc55a27 commit 8655d04

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kmeans_for_anchors.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def load_data(path):
150150

151151
cluster = cluster[np.argsort(cluster[:, 0] * cluster[:, 1])]
152152
print('avg_ratio:{:.2f}'.format(avg_iou(data, cluster)))
153-
print(data)
153+
print(cluster)
154154

155155
f = open("yolo_anchors.txt", 'w')
156156
row = np.shape(cluster)[0]

0 commit comments

Comments
 (0)