【YOLOv3】用自己的标注数据Kmeans聚类计算anchor

部署运行你感兴趣的模型镜像
代码地址: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

您可能感兴趣的与本文相关的镜像

Yolo-v5

Yolo-v5

Yolo

YOLO(You Only Look Once)是一种流行的物体检测和图像分割模型,由华盛顿大学的Joseph Redmon 和Ali Farhadi 开发。 YOLO 于2015 年推出,因其高速和高精度而广受欢迎

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值