arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1908.07678cs.CVcs.LG
收藏

Asymmetric Non-local Neural Networks for Semantic Segmentation

Huazhong University of Science and Technology(华中科技大学);University of Oxford(牛津大学)

Zhen Zhu, Mengde Xu, Song Bai, Tengteng Huang, Xiang Bai

英文摘要:

The non-local module works as a particularly useful technique for semantic segmentation while criticized for its prohibitive computation and GPU memory occupation. In this paper, we present Asymmetric Non-local Neural Network to semantic segmentation, which has two prominent components: Asymmetric Pyramid Non-local Block (APNB) and Asymmetric Fusion Non-local Block (AFNB). APNB leverages a pyramid sampling module into the non-local block to largely reduce the computation and memory consumption without sacrificing the performance. AFNB is adapted from APNB to fuse the features of different levels under a sufficient consideration of long range dependencies and thus considerably improves the performance. Extensive experiments on semantic segmentation benchmarks demonstrate the effectiveness and efficiency of our work. In particular, we report the state-of-the-art performance of 81.3 mIoU on the Cityscapes test set. For a 256x128 input, APNB is around 6 times faster than a non-local block on GPU while 28 times smaller in GPU running memory occupation. Code is available at: https://github.com/MendelXu/ANN.git.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑