(附代碼)CVPR2021|一個(gè)高效的金字塔切分注意力模塊PSA
點(diǎn)擊左上方藍(lán)字關(guān)注我們

論文:
https://arxiv.org/pdf/2105.14447v1.pdf
代碼:
https://github.com/murufeng/EPSANet
本文出發(fā)點(diǎn)

1. SE僅僅考慮了通道注意力,忽略了空間注意力。
2. BAM和CBAM考慮了通道注意力和空間注意力,但仍存在兩個(gè)最重要的缺點(diǎn):(1)沒(méi)有捕獲不同尺度的空間信息來(lái)豐富特征空間。(2)空間注意力僅僅考慮了局部區(qū)域的信息,而無(wú)法建立遠(yuǎn)距離的依賴。
3. 后續(xù)出現(xiàn)的PyConv,Res2Net和HS-ResNet都用于解決CBAM的這兩個(gè)缺點(diǎn),但計(jì)算量太大。
PSA



這里補(bǔ)充一下pyramid split attention中的pyramid。在《特征金字塔技術(shù)總結(jié)》中介紹了特征金字塔的兩種構(gòu)建方式,其中一種就是通過(guò)不同大小卷積核的卷積來(lái)構(gòu)建特征金字塔。因此,這里PSA中的Pyramid是由SPC模塊中的每組不同大小卷積核的卷積所構(gòu)建。
EPSANet



Conclusion

以ResNet-50和ResNet-101為backbone,加入各種注意力模塊的圖像識(shí)別效果對(duì)比


END
整理不易,點(diǎn)贊三連↓
評(píng)論
圖片
表情
