| 面向高分遥感影像道路提取的轻量级双注意力和特征补偿残差网络模型 |
| |
| 引用本文： | 陈振,陈芸芝,吴婷,李佳优.面向高分遥感影像道路提取的轻量级双注意力和特征补偿残差网络模型[J].地球信息科学,2022,24(5):949-961. |
| |
| 作者姓名： | 陈振 陈芸芝 吴婷 李佳优 |
| |
| 作者单位： | 1.福州大学 数字中国研究院（福建）,福州 3501082.卫星空间信息技术综合应用国家地方联合工程研究中心,福州 3501083.福州大学 空间数据挖掘与信息共享教育部重点实验室,福州 350108 |
| |
| 基金项目： | 国家自然科学基金项目(42071446);联通(福建)产业互联网有限公司委托项目(JC21-3502-2020-000559) |
| |
| 摘 要： | 针对高分辨率遥感影像背景复杂,道路提取容易受阴影、建筑物和铁路等背景信息干扰的问题,提出一种带有轻量级双注意力和特征补偿机制的DAFCResUnet模型。该模型在ResUnet的基础上,通过增加轻量级的双注意力和特征补偿模块实现模型在性能和时空复杂度上的平衡。其中,双注意力模块可以增强模型的特征提取能力,特征补偿模块可以融合网络中来自深浅层的道路特征。在DeepGlobe和GF-2道路数据集上的实验结果表明,DAFCResUnet模型的IoU和F1-score可以达到0.6713、0.8033和0.7402、0.8507,模型的整体精度优于U-Net、ResUnet和VNet模型。与U-Net和ResUnet模型相比,DAFCResUnet模型仅增加了少量的计算量和参数量,但IoU和F1-score均有较大幅度的提高;与VNet模型相比,DAFCResUnet模型在计算量和参数量远低于VNet的情况下取得了更高的精度,模型在精度和时空复杂度两方面均有优势。相比其他对比模型,DAFCResUnet模型具有更强的特征提取和抗干扰能力,能更好解决道路上的干扰物、与道路特征相似地物、树荫或阴影遮挡等造成的道路空洞、误提和漏提现象。
|
| 关 键 词： | 深度学习 道路提取 高分辨率遥感影像 残差网络 U-Net模型 双注意力模块 编解码器 特征补偿 |
| 收稿时间： | 2021-09-30 |
A Lightweight Dual Attention and Feature Compensated Residual Network Model for Road Extraction from High-Resolution Remote Sensing Images |
| |
| CHEN Zhen,CHEN Yunzhi,WU Ting,LI Jiayou.A Lightweight Dual Attention and Feature Compensated Residual Network Model for Road Extraction from High-Resolution Remote Sensing Images[J].Geo-information Science,2022,24(5):949-961. |
| |
| Authors: | CHEN Zhen CHEN Yunzhi WU Ting LI Jiayou |
| |
| Institution: | 1. The Academy of Digital China(Fujian), Fuzhou University, Fuzhou 350108, China2. National and Local Joint Engineering Research Center for the Comprehensive Application of Satellite Space Information Technology, Fuzhou 350108, China3. Key Laboratory of Spatial Data Mining and Information Sharing, Ministry of Education, Fuzhou University, Fuzhou 350108, China. |
| |
| Abstract: | Aiming at the problem that the background of high-resolution remote sensing images is complex and road extraction is easily disturbed by background information such as shadows, buildings, and railroads, the DAFCResUnet model with lightweight dual attention and feature compensation mechanism is proposed in this study. The model is based on ResUnet and achieves a balance between model performance and spatiotemporal complexity by adding lightweight dual attention and feature compensation modules. The dual attention module enhances the feature extraction capability of the model, and the feature compensation module can fuse the road features from deep and shallow layers in the network. The experimental results using DeepGlobe and GF-2 road datasets show that the IoU of the DAFCResUnet model can reach 0.6713, 0.8033, respectively, and the F1-score is 0.7402, 0.8507, respectively. The overall accuracy of the model is higher than that of U-Net, ResUnet, and VNet models. Compared with the U-Net and ResUnet models, the DAFCResUnet model only increases a small amount of computation and number of parameters, but the IoU and F1-score are improved substantially. Compared with the VNet model, the DAFCResUnet model achieves a higher accuracy with much lower computation and smaller number of parameters, and the model has advantages in both accuracy and spatiotemporal complexity. Compared with the other models, the DAFCResUnet model has stronger feature extraction and anti-interference ability, which can better solve the commission and omission caused by interfering objects on the road, ground features similar to roads, tree shade or shadow shading, etc. |
| |
| Keywords: | deep learning road extraction high-resolution remote sensing images residual network U-Net model dual attention module codecs feature compensation |
|
| 点击此处可从《地球信息科学》浏览原始摘要信息 |
| 点击此处可从《地球信息科学》下载免费的PDF全文 |
|
