CBAM

CBAM 我們提出了卷積塊注意力模塊(CBAM),這是一個簡單而有效的前饋卷積神經網絡的注意力模塊。給定一箇中間特徵圖,我們的模塊沿着通道和空間兩個獨立的維度依次推導注意力圖,然後將注意力圖乘以輸入的特徵圖,進行自適應的特徵細化。由於CBAM是一個輕量級的通用模塊,它可以無縫集成到任何CNN架構中,開銷可以忽略不計,並且可以和基礎CNN一起進行端到端訓練。我們通過在ImageNet-1K、MS C
相關文章
相關標籤/搜索