YOLOV7计算自己数据集的anchors。

m0_61880163

已于 2023-09-08 17:06:37 修改

阅读量607

点赞数 2

文章标签： YOLO

于 2023-09-08 17:02:51 首次发布

本文链接：https://blog.csdn.net/m0_61880163/article/details/132764002

版权

新建一个py文件，手动计算锚框：

import utils.autoanchor as autoAC
 
# 对数据集重新计算 anchors
new_anchors = autoAC.kmean_anchors('自己数据集的配置文件', 12, 640, 4.0, 1000, True) 
# n:anchor的总数  img_size：图像尺寸,初始值是640 　
#thr：默认设置是采用超参文档hyp.scratch.yaml里的“anchor_t”变量值；初始值是4.0
#gen：kmean聚类算法迭代次数，初始值是1000
#verbose：是否打印信息，初始值是true
print(new_anchors)