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个权重。 循环操作后,最终输出特征图
相关文章
相关标签/搜索