期刊
SIGNAL IMAGE AND VIDEO PROCESSING
卷 15, 期 5, 页码 1031-1039出版社
SPRINGER LONDON LTD
DOI: 10.1007/s11760-020-01828-8
关键词
Semantic segmentation; Fully convolutional network; Attention module
资金
- Joint fund for regional innovation and development of NSFC [U19A2083]
- Science and Technology Plan Project of Hunan Provinc [2016TP1020]
- Hunan Provincial Key Laboratory of Intelligent Information Processing and Application for Hengyang normal university [IIPA20K04]
The paper proposes a fully convolutional network with attention modules for semantic segmentation, which enhances pixel relevancy through post-processing and skip-layer attention, optimizes the loss function, and improves performance compared to other models.
Fully convolutional network is a powerful end-to-end model for semantic segmentation. However, it performs prediction pixel by pixel to pose weak consistency on intra-category. This paper proposes fully convolutional network with attention modules for semantic segmentation. Based on the framework of fully convolutional network, the post-processing attention module and skip-layer attention module are introduced to enhance the relevancy among pixels. Post-processing attention module is to calculate the similarity among pixels to obtain global information. Skip-layer attention module is designed to combine semantic information from a deep, coarse layer with contour information from a shallow, fine layer to produce the feature with high resolution and strong semantic information. Loss function, obtained by cross-entropy between estimated probability and label, is to optimize the network. Extensive experiments demonstrate that the proposed approach is superior to DeepLab and other models in performance of mean IoU with moderate computational complexity
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据