計算機視覺注意力機制

SENet 論文連接:https://arxiv.org/abs/1709.01507 通道域學習通道注意力,該網絡通過全局平均池化將特徵圖轉換爲1x1xc,然後通過壓縮通道數,再重構通道數學習網絡參數,最後通過sigmoid函數生成0~1之間的權重,與之前特徵圖進行元素相乘。下圖爲用在兩個模型中的示例。 CBAM 論文連接:https://arxiv.org/abs/1807.06521 通道注
相關文章
相關標籤/搜索