计算机科学与探索 ›› 2022, Vol. 16 ›› Issue (5): 1136-1145.DOI: 10.3778/j.issn.1673-9418.2105095

• 图形图像 • 上一篇    下一篇

全卷积注意力机制神经网络的图像语义分割

欧阳柳1, 贺禧1, 瞿绍军1,2,+()   

  1. 1.湖南师范大学 信息科学与工程学院,长沙 410081
    2.湖南师范大学 湖南湘江人工智能学院,长沙 410081
  • 收稿日期:2021-06-10 修回日期:2021-09-29 出版日期:2022-05-01 发布日期:2022-05-19
  • 通讯作者: + E-mail: qshj@hunnu.edu.cn
  • 作者简介:欧阳柳(1999—),男,湖北黄冈人,硕士研究生,CCF会员,主要研究方向为计算机视觉、深度学习。
    贺禧(2000—),男,湖南湘潭人,主要研究方向为计算机视觉、深度学习。
    瞿绍军(1979—),男,湖南永顺人,博士,高级实验师,CCF会员,主要研究方向为图像分割、计算机视觉、深度学习。
  • 基金资助:
    国家自然科学基金(12071126);湖南省教育厅科学研究项目(19C1149);国家级大学生创新创业训练计划项目(S202010542021)

Fully Convolutional Neural Network with Attention Module for Semantic Segmentation

OU Yangliu1, HE Xi1, QU Shaojun1,2,+()   

  1. 1. College of Information Science and Engineering, Hunan Normal University, Changsha 410081, China
    2. Hunan Xiangjiang Artificial Intelligence Academy, Hunan Normal University, Changsha 410081, China
  • Received:2021-06-10 Revised:2021-09-29 Online:2022-05-01 Published:2022-05-19
  • About author:OU Yangliu, born in 1999, M.S. candidate, member of CCF. His research interests include computer vision and deep learning.
    HE Xi, born in 2000. His research interests include computer vision and deep learning.
    QU Shaojun, born in 1979, Ph.D., senior experimentalist, member of CCF. His research interests include image segmentation, computer vision and deep learning.
  • Supported by:
    National Natural Science Foundation of China(12071126);Scientific Research Fund of Hunan Provincial Education Department(19C1149);National College Students’ Innovation and Entrepreneurship Training Program(S202010542021)

摘要:

全卷积神经网络是一种强大的端到端的模型,在语义分割领域应用广泛,获得了巨大的成功。研究人员提出了一系列基于完全卷积神经网络的方法,但是随着卷积和池化的持续性下采样,图像的上下文信息将会丢失,影响了像素级分类。针对完全卷积网络上下文信息丢失问题,提出基于像素的注意力方法。该方法利用计算高级特征图像素之间的联系来获取全局信息,增强像素之间的相关性,再结合空洞空间金字塔池化进一步提取图像的特征信息。针对图像的高层特征图像素丢失的问题,提出了基于图像不同层级的注意力方法。该方法将高层特征图中的信息作为指导对低层特征图中隐藏的信息进行挖掘,然后和高级特征图进行融合,充分利用高级特征图信息和低级特征图的信息。在实验方面,通过对比所提不同模块对全卷积神经网络分割性能的影响,验证了所提方法的有效性。同时在公认的图像语义分割数据集Cityscapes上与当前先进的网络进行实验对比,结果显示所提方法在客观评价指标和主观效果方面均存在优越性,并在Cityscapes官网测试集中达到了69.3%的准确率,性能比近期几个先进网络高出3~5个百分点。

关键词: 全卷积神经网络, 空洞空间金字塔池化, 注意力模型, 语义分割

Abstract:

A fully convolutional neural network is a powerful end-to-end model that is widely used in the field of semantic segmentation and has achieved great success. Researchers have proposed a series of methods based on a fully convolutional neural network. However, with the continuous subsampling of convolutions and pooling, the image contextual information will be lost, affecting the pixel-level classification. To solve the problem of context loss in a fully convolutional network, a pixel-based attention method is proposed, which calculates the relationship bet-ween high-level feature map pixels to obtain global information and enhance the correlation between pixels com-bined with atrous spatial pyramid pooling to further extract the image feature information. To solve the problem of pixel loss in the high-level feature map of an image, an attention method based on different levels of the image is proposed. This method uses the information in the high-level feature map as a guide to mine the hidden information in the low-level feature map and then fuses it with the high-level feature map to make full use of the high-level feature map and the low-level feature map information. In the experiment, the effectiveness of the proposed method is verified by comparing the effects of different modules on the segmentation results of a fully convolutional neural network. At the same time, experiments are carried out on the recognized image semantic segmentation dataset called Cityscapes and compared with the current advanced networks. The results show that the proposed method has advantages in both objective evaluation indicators and subjective effects, and achieves 69.3% accuracy in the Cityscapes official website test set. The performance is 3 to 5 percentage points higher than that of several recent advanced networks.

Key words: fully convolutional neural network, atrous spatial pyramid pooling, attention module, semantic segmentation

中图分类号: