节点文献
基于多任务学习的复杂城市遥感图像道路提取
Road Extraction from Complex Urban Remote Sensing Images Based on Multi-task Learning
【摘要】 提出一种新的遥感图像道路提取框架,旨在利用从道路边缘检测中获得的知识来提高道路提取的准确性。研究中引入了一个融合多尺度信息和视觉注意力机制的多尺度视觉注意力模块,并构建了一个级联特征融合模块以集成网络在不同尺度上的预测结果。在此基础上,构建了一个包含编码器和解码器的多尺度视觉注意网络(MSVANet)。同时,提出一个多任务学习框架,该框架结合了MSVANet,并采用粒子群优化算法对多任务学习框架的两个学习率超参数的自动选取进行优化。RNBD数据集的训练和测试结果表明,所提方法在各种分割精度指标和泛化能力方面均优于其他道路提取方法。
【Abstract】 In this paper,we propose a new framework for road extraction from remote sensing images that aims to utilize the knowledge gained from road edge detection to improve the accuracy of road extraction.A multi-scale visual attention module that fuses multi-scale information and visual attention mechanisms is introduced in the study,and a cascading feature fusion module is constructed to integrate the network’s prediction results at different scales.Based on this,we construct a multiscale visual attention network(MSVANet)containing encoders and decoders.A multi-task learning framework that incorporates the MSVANet is also proposed,and a particle swarm optimization algorithm(PSO)is used to optimize the automatic selection of the two learning rate hyperparameters of the multi-task learning framework.The training and testing results on the RNBD dataset show that the proposed method outperforms other road extraction methods in terms of various segmentation accuracy metrics and generalization ability.
【Key words】 Deep learning; Remote sensing image road extraction; Multi-task learning; Multi-scale visual attention network;
- 【文献出处】 计算机科学 ,Computer Science , 编辑部邮箱 ,2024年S2期
- 【分类号】TP751;U491
- 【下载频次】142