基于双重注意力特征增强网络的语义分割方法
Dual Attention Based Feature Enhanced Networks for Semantic Segmentation
摘要: 语意分割作为计算机视觉领域的研究热点之一,在地理信息系统、医疗影像分析和机器人等领域有广泛应用。然而现有的语义分割方法主要面临两个挑战,即类内不一致和类间难区分问题。为此,我们提出了一种基于双重注意力特征增强网络的方法来实现语义分割。该方法采用位置注意力模块与通道注意力模块来获取丰富的空间信息与上下文信息,并且在网络末端添加金字塔池化模块来聚合不同区域的上下文信息,提高网络捕获全局信息的能力。最终在标准数据集上的实验结果验证了本文方法的有效性。
Abstract: As one of the research hotspots in the field of computer vision, semantic segmentation has been widely applied in various fields such as geographic information systems, medical image analysis and robotics. However, contemporary semantic segmentation tasks generally face two challenges, namely intra-class inconsistency problem and inter-class indistinction problem. To this end, we solve the semantic segmentation by proposing Dual Attention based Feature Enhanced Networks. In this method, the position attention module and channel attention module are used to obtain rich spatial and context information, and the pyramid pooling module is added at the end of the network to aggregate the context information of different regions, which could improve the capability of the networks to capture global information. Finally, the experimental results on the standard dataset demonstrate the effectiveness of the proposed method.
文章引用:赵芮, 于晓艳, 荣宪伟. 基于双重注意力特征增强网络的语义分割方法[J]. 计算机科学与应用, 2020, 10(11): 1944-1951. https://doi.org/10.12677/CSA.2020.1011205

参考文献

[1] Garcia-Garcia, A., et al. (2017) A Review on Deep Learning Techniques Applied to Semantic Segmentation. Internation-al Conference on Computational Linguistics, Spain, 22 April 2017, 2132-2144.
[2] 陈一鸣, 彭艳兵, 高剑飞. 基于深度学习的遥感图像新增建筑物语义分割[J]. 计算机与数字工程, 2019, 47(12): 3182-3186.
[3] Long, J., Shel-hamer, E. and Darrell, T. (2014) Fully Convolutional Networks for Semantic Segmentation. IEEE Transactions on Pat-tern Analysis & Machine Intelligence, 39, 640-651. [Google Scholar] [CrossRef
[4] Yu, F. and Koltun, V. (2016) Multi-Scale Context Aggregation by Dilated Convolutions.
[5] Wang, X., Girshick, R., Gupta, A., et al. (2018) Non-Local Neural Networks. IEEE Computer Society Conference on Computer Vision and Pattern Recogni-tion, Salt Lake City, 18-22 June 2018, 7794-7803. [Google Scholar] [CrossRef
[6] Huang, Z., Wang, X., Huang, L., et al. (2019) CCNet: Criss-Cross Attention for Semantic Segmentation. IEEE International Conference on Computer Vision, Seoul, 27-28 October 2019, 603-612. [Google Scholar] [CrossRef
[7] Chao, P., et al. (2017) Large Kernel Matters—Improve Semantic Segmentation by Global Convolutional Network. IEEE Conference on Computer Vision and Pattern Recognition, Hono-lulu, 21-26 July 2017, 1743-1751. [Google Scholar] [CrossRef
[8] Chen, L., et al. (2020) ANU-Net: Attention-Based Nested U-Net to Exploit Full Resolution Features for Medical Image Segmentation. Computers & Graphics, 90, 11-20. [Google Scholar] [CrossRef
[9] Ronneberger, O., Fischer, P. and Brox, T. (2015) U-Net: Convolu-tional Networks for Biomedical Image Segmentation. International Conference on Medical Image Computing and Com-puter-Assisted Intervention, Munich, 5-9 October 2015, 234-241. [Google Scholar] [CrossRef
[10] Yu, C., et al. (2018) BiSeNet: Bilateral Segmentation Network for Real-Time Semantic Segmentation. European Conference on Computer Vision, Munich, 8-14 September 2018, 334-349. [Google Scholar] [CrossRef
[11] Yu, C., Wang, J., Peng, C., et al. (2018) Learning a Discriminative Feature Network for Semantic Segmentation. IEEE Computer Society Conference on Computer Vision and Pattern Recognition, Salt Lake City, 18-22 June 2018, 1857-1866. [Google Scholar] [CrossRef
[12] Zhao, H., Shi, J., Qi, X., et al. (2016) Pyramid Scene Parsing Network. IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, 27-30 June 2016, 6230-6239. [Google Scholar] [CrossRef
[13] 翟鹏博, 杨浩, 宋婷婷. 结合注意力机制的双路径语义分割[J]. 中国图象图形学报, 2020, 25(8): 1627-1636.
[14] Lin, T.Y., Goyal, P., Girshick, R., et al. (2017) Focal Loss for Dense Object Detection. IEEE Transactions on Pattern Analysis & Machine Intelligence, 99, 2999-3007.
[15] Hari-haran, B., Arbelaez, P., Bourdev, L.D., et al. (2011) Semantic Contours from Inverse Detectors. IEEE International Conference on Computer Vision, Barcelona, 6-13 November 2011, 991-998. [Google Scholar] [CrossRef
[16] Krizhevsky, A., Sutskever, I. and Hinton, G.E. (2012) Imagenet Classification with Deep Convolutional Neural Networks. Advances in Neural Information Processing Systems, Vol. 2, 1097-1105.
[17] 熊炜, 童磊, 金靖熠. 基于卷积神经网络的语义分割算法研究[J/OL]. 计算机应用研究, 2020, 38(3): 1-5.
[18] Russakovsky, O., Deng, J., Su, H., et al. (2015) ImageNet Large Scale Visual Recognition Challenge. International Journal of Computer Vision, 115, 211-252. [Google Scholar] [CrossRef
[19] Liu, W., Rabinovich, A. and Berg, A.C. (2015) ParseNet: Looking Wider to See Better. arXiv preprint arXiv: 1506.04579
[20] Chen, L.C., Papandreou, G., Schroff, F., et al. (2017) Rethinking Atrous Convolution for Se-mantic Image Segmentation.
[21] Zhang, H., Dana, K., Shi, J., et al. (2018) Context Encoding for Semantic Segmenta-tion. IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, 18-23 June 2018, 7151-7160. [Google Scholar] [CrossRef
[22] Zhang, Z., Zhang, X., Peng, C., et al. (2018) ExFuse: Enhancing Feature Fusion for Semantic Segmentation. European Conference on Computer Vision, Munich, 8-14 September 2018, 269-284. [Google Scholar] [CrossRef
[23] Jun, F., et al. (2019) Stacked Deconvolutional Network for Semantic Segmentation. International Conference on Image Processing, Taipei, 22-25 September 2019, 3085-3089. [Google Scholar] [CrossRef
[24] Luo, P., Wang, G., Lin, L., et al. (2017) Deep Dual Learning for Semantic Image Segmentation. IEEE International Conference on Computer Vision, Venice, 22-29 October 2017, 2737-2745. [Google Scholar] [CrossRef
[25] Xia, L., Zhong, Z.S., Wu, J.L., et al. (2019) Expecta-tion-Maximization Attention Networks for Semantic Segmentation. IEEE International Conference on Computer Vision, Seoul, 27-28 October 2019, 9166-9175.