关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比

首先在开头下一个定义。

此处以下所指的模型1为未做任何改动之前的yolov5-6.0网络,模型2为将yolov5网络中的骨干层里C3模块全部替换为注意力模块的网络,其他部分未做改动,不知道怎么加的看这里yolov5加入CBAM,SE,CA,ECA注意力机制,纯代码_zqt321的博客-CSDN博客

这是原图

 

先来看模型2的检测结果,我一直在做工地的安全帽检测,

v5+SE:

 重点关注我在检测结果中画蓝线的圈圈。

v5+CBAM:

V5+ECA: 

 V5+CA:

再来看模型1的检测结果:

yolov5-6.0

 总结:

因数据集涉及到公司机密,其实涉及到机密的那一部分更加能体现出来两者之间的差别,总结也就一句话,不要试图和一屋子的博士比聪明,他们把好用的trick都放一块了,用就行了。在v5刚出来的时候很多人不同意v5这个称号,但是从现在v5的6.0版本使用效果来看,个人认为已经极大的满足了工程上的使用,并且精度也比5.0版本有所提升,yolov5真香!

最后祝大家早日年薪百万!

猜你喜欢

转载自blog.csdn.net/zqt321/article/details/121796047
今日推荐