詳解五種即插即用的視覺(jué)注意力模塊

SE注意力模塊

SE注意力模塊的全稱(chēng)是Squeeze-and-Excitation block、其中Squeeze實(shí)現(xiàn)全局信息嵌入、Excitation實(shí)現(xiàn)自適應(yīng)權(quán)重矯正，合起來(lái)就是SE注意力模塊。

把SE注意力模塊添加到Inception Module中與ResNet的殘差Module中，實(shí)現(xiàn)SE注意力版本的Inception與ResNet模型

SE注意力模塊集成殘差的四種不同結(jié)構(gòu)設(shè)計(jì)

CA注意力模塊

圖中最右側(cè)是CA注意力模塊、左側(cè)是SE注意力模塊、中間是CBAM注意力模塊、CA注意力通過(guò)坐標(biāo)信息嵌入和坐標(biāo)注意力生成兩個(gè)步驟把通道關(guān)系和長(zhǎng)程依賴(lài)關(guān)系精確編碼。

CBAM注意力模塊

該圖顯示了如何把一個(gè)CBAM模塊集成到到一個(gè)殘差模塊中，實(shí)現(xiàn)殘差CBAM模塊，CBAM是集成了通道注意力與空間注意力、通道和空間注意力模塊生成首先是通過(guò)平均池化和最大池化分別生成通道與空間注意力的輸入，然后通過(guò)MLP與7x7卷積輸出通道子模塊和空間子模塊。

ECA注意力模塊

通過(guò)全局均值池化(GAP)獲得的聚合通道特征，然后通過(guò)大小為k濾波器的快速1D卷積，然后重新構(gòu)建一個(gè)維的卷積輸出結(jié)果，乘以輸入的每個(gè)通通道數(shù)據(jù)作為輸出實(shí)現(xiàn)注意力機(jī)制為ECA。

GAM注意力模塊

全局注意力模塊基于CBAM注意力模塊集成上重新設(shè)計(jì)通道與空間兩個(gè)子模塊實(shí)現(xiàn)，其中通道注意力直接使用MLP實(shí)現(xiàn)、空間注意力通過(guò)兩個(gè)7乘7的卷積實(shí)現(xiàn)，最終GAM注意力添加到ResNet網(wǎng)絡(luò)的效果相比SE注意力、CBAM注意力有明顯提升。

YOLOv57.0 注意力加持版本

審核編輯：劉清