激光与光电子学进展, 2020, 57 (4): 041015, 网络出版: 2020-02-20   

基于注意力机制的遥感图像分割模型 下载: 1915次

Remote Sensing Image Segmentation Model Based on Attention Mechanism
作者单位
陕西师范大学计算机科学学院, 陕西 西安 710119
摘要
针对遥感图像目标密集、尺度不一、存在遮挡等特点,提出一种基于注意力机制的遥感图像分割模型用于目标分割。该模型建立在深度图像分割模型的基础上,提出在高低层特征融合之前采用通道注意力机制对低层特征进行加权处理,增强目标特征并抑制背景特征,提高信息的融合效率。为进一步增强模型对目标特征的响应能力,提出位置注意力机制对解码阶段最后的特征进行处理。最后,将加权融合后的特征图上采样到原图大小并预测像素类别。在两个遥感道路数据集上进行实验并与相关模型进行比较,结果表明所提模型在遥感影像道路提取中性能优异,可应用于复杂的遥感影像目标分割。
Abstract
Aim

ing at the remote sensing images with target intensive, multi-scale, and occlusion, a remote sensing image segmentation model based on the attention mechanism is proposed herein. The proposed method is based on the deep image segmentation model. The channel attention mechanism is used for weighting the low-level features before high-low layer feature fusion, thus enhancing the target features, suppressing the background features, and improving the information fusion efficiency. A positional attention mechanism is proposed to process the final features of the decoding phase for further enhancing the responsiveness of the model to the target features. At last, weighted and aggregated feature maps are up-sampled to the original image size for pixel label prediction. Experiments on two remote sensing road datasets and comparisons with related models show that the proposed model displays excellent performance in remote sensing image road extraction and can be employed to complex remote sensing image segmentation.

刘航, 汪西莉. 基于注意力机制的遥感图像分割模型[J]. 激光与光电子学进展, 2020, 57(4): 041015. Hang Liu, Xili Wang. Remote Sensing Image Segmentation Model Based on Attention Mechanism[J]. Laser & Optoelectronics Progress, 2020, 57(4): 041015.

本文已被 4 篇论文引用
被引统计数据来源于中国光学期刊网
引用该论文: TXT   |   EndNote

相关论文

加载中...

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!