CCNet:Criss-Cross Attention for Semantic Segmentation

[code pytorch](t https://github.com/speedinghzl/CCNet) 兩種基於注意力的上下文aggregation圖 對於每個位置(例如藍色),Non-local模塊都會生成密集的注意力圖,該圖的權重爲H×W(綠色)。 對於每個位置(例如藍色),criss-cross注意模塊會生成一個稀疏的注意圖,該圖僅具有H + W-1個權重。 循環操作後,最終輸出特徵圖
相關文章
相關標籤/搜索