YOLOv5改进之添加CBAM注意力机制的方法
前 言：
作为当前先进的深度学习目标检测算法YOLOv5，已经集合了大量的trick，但是在处理一些复杂背景问题的时候，还是容易出现错漏检的问题。此后的系列文章，将重点对YOLOv5的如何改进进行详细的介绍，目的是为了给那些搞科研的同学需要创新点或者搞工程项目的朋友需要达到更好的效果提供自己的微薄帮助和参考。
解决问题：
加入CBAM双通道注意力机制，可以让网络更加关注待检测目标，提高检测效果
添加方法：
第一步：确定添加的位置，作为即插即用的注意力模块，可以添加到YOLOv5网络中的任何地方。本文以添加进卷积Conv模块中为例。
第二步：common.py构建融入CBAM模块的Conv_CBAM,与原Conv模块不同的是：在该模块中的激活函数后加入CBAM模块。
class Conv_CBAM(nn.Module):
# Standard convolution
def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True): # ch_in, ch_out, kernel, stride, padding, groups
super(Conv_CBAM, self).__init__()
self.conv = nn.Conv2d(c1, c2, k, s, autopad(k, p), groups=g, bias=False)
self.bn = nn.BatchNorm2d(c2)
self.act = nn.Hardswish() if act else nn.Identity()
self.ca = ChannelAttention(c2)
self.sa = SpatialAttention()
def forward(self, x):
x = self.act(self.bn(self.conv(x)))
x = self.ca(x) * x
x = self.sa(x) * x
return x
def fuseforward(self, x):
return self.act(self.conv(x))第三步：yolo.py中注册我们进行修改的Conv_CBAM模块
第四步：修改yaml文件，本文以修改主干特征提取网络为例，将原Conv模块改为Conv_CBAM即可。
第五步：将train.py中改为本文的yaml文件即可，开始训练。
结 果：
本人在多个数据集上做了大量实验，针对不同的数据集效果不同，同一个数据集的不同添加位置方法也是有差异，需要大家进行实验。有效果有提升的情况占大多数。
PS：CBAM双通道注意力机制，不仅仅是可以添加进YOLOv5，也可以添加进任何其他的深度学习网络，不管是分类还是检测还是分割，主要是计算机视觉领域，都可能会有不同程度的提升效果。
总 结
到此这篇关于YOLOv5改进之添加CBAM注意力机制的文章就介绍到这了,更多相关YOLOv5添加CBAM注意力机制内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
Python处理文本的功能非常强大，但是如果是初学者，没有搞清楚python中的编码机制，也经常会遇到乱码或者decode error。本文的目的是简明扼要地说明python的编码机制，并给出一些建议,需要的朋友可以参考下2018-10-10
使用urllib2模块进行基于url的HTTP请求等操作大家也许都比较熟悉,这里我们再深入来了解一下urllib2针对HTTP的异常处理相关功能,一起来看一下Python网络编程中urllib2模块的用法总结:2016-07-07
纯Python开发的nosql数据库CodernityDB介绍和使用实例
这篇文章主要介绍了纯Python开发的nosql数据库CodernityDB介绍和使用实例,本文实例包含数据插入、数据更新、数据删除、数据查询等,需要的朋友可以参考下2014-10-10
使用Pycharm(Python工具)新建项目及创建Python文件的教程
这篇文章主要介绍了使用Pycharm(Python工具)新建项目及创建Python文件的教程,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-04-04
这篇文章主要介绍了Python的MongoDB模块PyMongo操作方法集锦,包括数据的增删查改以及索引等相关的基本操作,需要的朋友可以参考下2016-01-01
import是我们经常使用的功能，方法大家也都会。虽然用的多，但它的本质我们似乎没有好好的了解过。本文就import的本质解析，向大家作了介绍，需要的朋友可以参考下。2017-10-10
解决Django删除migrations文件夹中的文件后出现的异常问题
今天小编就为大家分享一篇解决Django删除migrations文件夹中的文件后出现的异常问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
在本篇文章中小编给大家分享了关于自学python的建议和周期预算，有兴趣的朋友们可以学习参考下。2019-01-01
这篇文章主要介绍了Python 实现随机数详解及实例代码的相关资料,需要的朋友可以参考下2017-04-04
本文主要分享了Python交通数据分析与可视化的实战！其中主要是使用TransBigData库快速高效地处理、分析、挖掘出租车GPS数据，感兴趣的可以了解一下2023-04-04
最新评论