结合掩码图像建模和对比学习的道路提取方法.pdfVIP

第37卷第4期系统仿真学报系统仿真学报©Vol.37No.4

2025年4月JournalofSystemSimulationApr.2025

结合掩码图像建模和对比学习的道路提取方法结合掩码图像建模和对比学习的道路提取方法

*

伍江江，李政宏，沙志超，陈浩，彭双，杜春，李军

(国防科技大学电子科学学院，湖南长沙410073)

摘要摘要：：针对遥感影像道路提取的遮挡问题，提出了结合掩码图像建模和对比学习的道路提取方法，

其模型训练过程包括掩码预训练阶段和对比训练阶段。掩码预训练阶段主要进行掩码图像重建，

训练模型从部分区域被随机遮挡的图像恢复出整张图像；对比训练阶段主要对于预测错误以及置

信度较低的区域进行对比学习，拉近相同类别对应的特征距离而增大不同类别特征间的距离。实

验结果验证了方法的有效性和可用性。

关键词关键词：：遥感影像；道路提取；掩码图像建模；对比学习；图像重建

中图分类号：TP751.1文献标志码：A文章编号：1004-731X(2025)04-0922-11

DOI:10.16182/j.issn1004731x.joss.24-1083

引用格式引用格式:伍江江,李政宏,沙志超,等.结合掩码图像建模和对比学习的道路提取方法[J].系统仿真学报,2025,37

(4):922-932.

Referenceformat:WuJiangjiang,LiZhenghong,ShaZhichao,etal.AMethodforRoadExtractionUsingMaskedImage

ModelingandContrastiveLearning[J].JournalofSystemSimulation,2025,37(4):922-932.

AMethodforRoadExtractionUsingMaskedImageModelingandContrastiveLearning

*

WuJiangjiang,LiZhenghong,ShaZhichao,ChenHao,PengShuang,DuChun,LiJun

(CollegeofElectronicScienceandTechnology,NationalUniversityofDefenseTechnology,Changsha410073,China)

Abstract:Aimingattheocclusionproblemofroadextractionfromremotesensingimages,aroad

extractionmethodcombiningMIMandCLisproposed,themodeltrainingprocessincludesamaskedpre-

trainingstageandacontrasttrainingstage.Themaskedpre-trainingstagemainlycarriesoutmaskimage

reconstruction,andtrainsthemodeltorecoverthewholeimagefromsomeareasthatarerandomly

occluded.Thecomparisontrainingstageismainlyforthep