Skip to content

Commit

Permalink
Update kmeans_for_anchors.py
Browse files Browse the repository at this point in the history
  • Loading branch information
bubbliiiing authored Jan 15, 2022
1 parent ddf9d16 commit 02b0e1f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion kmeans_for_anchors.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit 02b0e1f

Please sign in to comment.