光电子·激光2024,Vol.35Issue(1) :51-58.DOI:10.16136/j.joel.2024.01.0553

基于iHDODC-LinkNet网络的遥感图像道路提取方法

Remote sensing image road extraction method based on iHDODC-LinkNet network

陈国军 朱燕宁 耿润田 李子祥
光电子·激光2024,Vol.35Issue(1) :51-58.DOI:10.16136/j.joel.2024.01.0553

基于iHDODC-LinkNet网络的遥感图像道路提取方法

Remote sensing image road extraction method based on iHDODC-LinkNet network

陈国军 1朱燕宁 1耿润田 1李子祥1
扫码查看

作者信息

  • 1. 中国石油大学(华东)计算机科学与技术学院,山东青岛 266580
  • 折叠

摘要

遥感图像的道路提取在推动城乡发展规划及建设方面具有重要意义.然而,目前传统方法对于遥感图像道路提取存在工程量大、效率低下的问题,基于深度学习的方法又存在复杂场景下提取精度不高和连通性差等问题.针对上述存在的问题,为提高不同地貌区域的道路提取精度,本文提出一种基于iHDODC-LinkNet网络的高分辨率遥感图像道路提取方法.该方法在语义分割模型D-LinkNet的基础上进行改进:一方面使用ResNeSt50重建D-LinkNet网络并添加预训练模型,提出一种混联深度过参数化扩张卷积(hybrid depthwise over-parameterized dilated convolu-tion,HDODC)模块;另一方面采用迭代注意力特征融合(iterative attentional feature fusion,iAFF)机制替换原始的相加融合,从而使模型关注于道路的全局信息.最后,在马萨诸塞州道路数据集和某省高速公路场景数据集上进行训练并通过测试集的提取效果证明模型改进方法的有效性.根据实验模型分割效果表明,改进后的方法在测试集上F1达到71.66%,相比原始模型提升了10%,能够得到效果更好的分割结果.

Abstract

Road extraction from remote sensing images is of great significance in promoting urban and ru-ral development planning and construction.However,the traditional methods for road extraction from re-mote sensing images have the problems of large engineering quantities and low efficiency,and the meth-ods based on depth learning have the problems of low extraction accuracy and poor connectivity in com-plex scenes.To solve the above problems and improve the accuracy of road extraction in different geo-morphic regions,this paper proposes a road extraction method based on iHDODC LinkNet network for high-resolution remote sensing images.This method is improved on the basis of the semantic segmenta-tion model D-LinkNet:on the one hand,ResNeSt50 is used to reconstruct the D-LinkNet network and a pre training model is added to propose a hybrid depthwise over-parameterized dilated convolution(HDODC)module.On the other hand,iterative attentional feature fusion(iAFF)mechanism is used to replace the original additive fusion,so that the model focuses on the global information of the road.Final-ly,the training is carried out on the Massachusetts road dataset and a provincial highway scene dataset,and the effectiveness of the improved model is proved by the extraction effect of the test set.According to the experimental model segmentation effect,the improved method applied to F1 reaches 71.66%,which is 10%higher than the original model,and better segmentation results can be obtained.

关键词

语义分割/连通性/混联深度过参数化扩张卷积(HDODC)/注意力特征融合

Key words

semantic segmentation/connectivity/hybrid depthwise over-parameterized dilated convolu-tion(HDODC)/attention feature fusion

引用本文复制引用

基金项目

山西省交通建设科技项目(2019-2-8)

出版年

2024
光电子·激光
天津理工大学 中国光学学会

光电子·激光

CSCD北大核心
影响因子:1.437
ISSN:1005-0086
被引量1
参考文献量17
段落导航相关论文