YOLOv7 训练前手动计算锚定框

使用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的种子（初始点）是随机的，所以生成的锚框都不一样，多生成几次试试看。