代码地址:https://github.com/lars76/kmeans-anchor-boxes
一、计算出归一化的anchor
1.修改上述代码中的example.py
ANNOTATIONS_PATH = '……' # 存放xml标注文件的路径
CLUSTERS = 9 # YOLOv3对应9个anchor
2.运行example.py即可得到归一化的anchor
输出如下的结果:
Accuracy: xx%
Boxes:
[[0.24117647 0.55191257] # 每一个 anchor的宽/图像的宽 ,高/图像的高
[0.43026205 0.36341757]
[……]
[……]
[……]
[……]
[……]
[……]
[0.12742382 0.37146371]]
Ratios:
[0.15, 0.34, 0.43, 0.44 …… ]
二、进一步调整
1.根据YOLOv3所使用的图片大小得到实际的anchor大小
比如cfg文件的参数如下所示:

则anchor在上文的基础上×416即可
2.排序并取整
YOLOv3中默认使用的anchors都是整数,我们也可调整为相同的形式

参考:
https://blog.youkuaiyun.com/weixin_42419002/article/details/100566926
67

被折叠的 条评论
为什么被折叠?



