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

首先在开头下一个定义。
此处以下所指的模型1为未做任何改动之前的yolov5-6.0网络,模型2为将yolov5网络中的骨干层里C3模块全部替换为注意力模块的网络,其他部分未做改动,不知道怎么加的看这里yolov5加入CBAM,SE,CA,ECA注意力机制,纯代码_zqt321的博客-CSDN博客)
这是原图
深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片

深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片



先来看模型2的检测结果,我一直在做工地的安全帽检测,
v5+SE: 深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片


深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片

重点关注我在检测结果中画蓝线的圈圈。
v5+CBAM: 深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片

【深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比】深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片


V5+ECA:深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片


深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片

V5+CA: 深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片


深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片


再来看模型1的检测结果:
yolov5-6.0 深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片

深度学习|关于添加注意力模块的yolov5-5.0与yolov5-6.0的检测效果对比
文章图片

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

    推荐阅读