首页|基于编解码网络的自动道路提取框架

基于编解码网络的自动道路提取框架

扫码查看
原文链接
  • 国家科技期刊平台
  • NETL
  • NSTL
  • 万方数据
基于深度学习的道路提取是近十年来语义分割的热点之一。本研究提出一种基于编解码网络的遥感图像道路自动提取框架。首先，将预训练好的ResNet34网络迁移到U-Net网络并替换其编码结构，加深网络层数，降低道路分割的错误率和细节丢失。其次，利用通道注意力机制，对编码器上采样得到的特征图进行信息选择，将目标特征进行权重优化，在增强图像目标区域特征的同时抑制背景及噪声区域，优化背景复杂度高的遥感图像特征提取效果。最后，提出一种自适应sigmoid损失函数，解决道路和背景占比不平衡的问题，使模型在客观程度上达到最优解。实验结果表明，与其他几种语义分割网络方法相比。本研究所提的方法有效降低了道路提取的错误率，提高了道路图像的分割精度。
Automatic road extraction framework based on codec network
Road extraction based on deep learning is one of hot spots of semantic segmentation in the past decade.In this work,we proposed a framework based on codec network for automatic road extraction from remote sensing images.Firstly,a pre-trained ResNet34 was migrated to U-Net and its encoding structure was replaced to deepen the number of network layers,which reduces the error rate of road segmentation and the loss of details.Secondly,dilated convolution was used to connect the encoder and the decoder of network to expand the receptive field and retain more low-dimensional information of the image.Afterwards,the channel attention mechanism was used to select the information of the feature image obtained by up-sampling of the encoder,the weights of target features were optimized to enhance the features of target region and suppress the features of background and noise regions,and thus the feature extraction effect of the remote sensing image with complex background was optimized.Finally,an adaptive sigmoid loss function was proposed,which optimizes the imbalance between the road and the background,and makes the model reach the optimal solution.Experimental results show that compared with several semantic segmentation networks,the proposed method can greatly reduce the error rate of road segmentation and effectively improve the accuracy of road extraction from remote sensing images.

remote sensing imageroad extractionResNet34U-Netchannel attention mechanismsigmoid loss function

王霖、沈瑜、张泓国、梁栋、牛东兴

展开 >

兰州交通大学电子与信息工程学院,甘肃兰州 730070

中铁科学研究院有限公司,四川成都 610036

遥感图像 道路提取 ResNet34 U-Net 通道注意力机制 sigmoid损失函数

National Natural Science Foundation of China2021 Longyuan Youth Innovation and Entrepreneurship Talent(Team)Young Doctoral Fund of Higher Education Institutions of Gansu ProvinceEmployment and Entrepreneurship Improvement Project of University Students of Gansu ProvinceIntelligent Tunnel Supervision Robot Research Project(China Railway Scientific Research Institute Scientific Research)Lanzhou Jiaotong University Youth FoundationGansu Higher Education Research ProjectGansu Dunhuang Cultural Relics Protection Research Center Open Project

618640252021QB-492021-C-1232020-KJ016-Z016-A220150052016A-018GDW2021YB15

2024

10.62756/jmsi.1674-8042.2024033

测试科学与仪器
中北大学

测试科学与仪器

影响因子：0.111
ISSN：1674-8042
年,卷(期)：2024.15(3)