Asymmetric Non-local Neural Networks for Semantic Segmentation

August 21, 2019 · Entered Twilight · 🏛 IEEE International Conference on Computer Vision

"Last commit was 6.0 years ago (≥5 year threshold)"

Evidence collected by the PWNC Scanner

Repo contents: .gitignore, LICENSE, README.md, data.conf, datasets, demos, docs, efficiency_statics, extensions, hypes, imagesite, main.py, methods, metrics, models, requirements.txt, resource, scripts, utils

Authors Zhen Zhu, Mengde Xu, Song Bai, Tengteng Huang, Xiang Bai arXiv ID 1908.07678 Category cs.CV: Computer Vision Cross-listed cs.LG Citations 676 Venue IEEE International Conference on Computer Vision Repository https://github.com/MendelXu/ANN.git ⭐ 313 Last Checked 2 months ago

Abstract

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.