语义分割中的nonlocal[5]-Interlaced Sparse Self-Attention for Semantic Segmentation

https://arxiv.org/abs/1907.12273
这篇文章依然是通过将non-local中费时费力的affinity map分解为多个便于得到的小affinity map得到加速效果

这个图形象的说明了本文将dense的两两之间的关系，转换为了两步：long range和short range
具体如下

实现代码如下：

效果如何呢：