注意力机制论文:CCNet: Criss-Cross Attention for Semantic Segmentation及其PyTorch实现

CCNet: Criss-Cross Attention for Semantic Segmentation
PDF: https://arxiv.org/pdf/1811.11721.pdf
PyTorch代码: https://github.com/shanglianlm0525/PyTorch-Networks

提出了一个CCNet,具有以下优点

• 1）较少的GPU内存使用;
• 2）计算效率高;
• 3）最先进的性能。

1 Criss-Cross Attention

2 Recurrent Criss-Cross Attention

3 CCNet

4 实验结果

PyTorch代码: