使用utils/autoanchor.py生成自己数据集匹配的锚框。
新建一个py文件
import utils.autoanchor as autoAC
# 对数据集重新计算 anchors
new_anchors = autoAC.kmean_anchors('./data/insulator.yaml', 12, 640, 5.0, 1000, True)
print(new_anchors)
由于每次kmeans的种子(初始点)是随机的,所以生成的锚框都不一样,多生成几次试试看。