(附代碼)CVPR2021|一個高效的金字塔切分注意力模塊PSA
點擊左上方藍字關注我們

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

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



這里補充一下pyramid split attention中的pyramid。在《特征金字塔技術總結》中介紹了特征金字塔的兩種構建方式,其中一種就是通過不同大小卷積核的卷積來構建特征金字塔。因此,這里PSA中的Pyramid是由SPC模塊中的每組不同大小卷積核的卷積所構建。
EPSANet



Conclusion

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


END
整理不易,點贊三連↓
評論
圖片
表情
