日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后，達者為師

網站首頁 編程語言 正文

YOLOv5改進之添加CBAM注意力機制的方法_python

作者：人工智能算法研究院 ? 更新時間： 2022-12-15 編程語言

前 言：

作為當前先進的深度學習目標檢測算法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，也可以添加進任何其他的深度學習網絡，不管是分類還是檢測還是分割，主要是計算機視覺領域，都可能會有不同程度的提升效果。

總 結

原文鏈接：https://blog.csdn.net/m0_70388905/article/details/125892144

欄目分類
最近更新