【目标检测】K-means++计算anchors【附代码】

本文介绍了如何使用K-means++算法为YOLO目标检测模型生成Anchors。作者在尝试不同数据集的K-means++实现后,调试并整合了代码,提供了一个可行的解决方案。文章中引用了多个CSDN博客和GitHub资源作为参考。
摘要由CSDN通过智能技术生成

昨天运行yolo系列模型时,先用kmeans聚类得到了anchor,然后训练模型。训练结束后想用kmeans++或者其他改进版,重新聚类后得到anchor再训练模型,做一个结果对比。

于是在网上各种找资料、找源码。。。此处想啰嗦几句,我知道kmeans++的算法思想,手中也有kmeas算法聚类得到anchor的源码,见我上次写的这篇博客,但是,总有一种依赖的思想,看看网上能不能找到源码,实在是不想写(其实是自己太菜。。。留下无情的泪水)。

结果总是残酷的,找了一下午,都没有找到合乎我心意的源码(那种copy-paste过来直接能用的代码)。将网上寻觅来的代码经过一番debug,终于实现了kmeans++聚类数据得到anchor,哈哈,由于代码风格的不同,yolo数据集也不相同(殊途同归)因此在kmeans++实现上也不相同(此处指的是代码风格,输入接口,对于功能和思想是相同的)。

总共调试运行了针对3种不同数据集的kmeans++算法(做个记录,希望以后少走弯路,提高自身的coding能力&

评论 66
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

机器不学习我学习

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值