一种基于多维多尺度U-net网络的遥感图像道路提取方法

摘要

本发明请求保护一种基于多维多尺度U‑net网络的遥感图像道路提取方法，首先将处理后的数据输入编码网络，在编码网络部分结合残差结构和注意力特征融合机制对道路信息进行提取，综合考虑全局和局部的特征提取，提高网络对道路细节目标的提取。其次，在桥接网络部分加入空洞空间金字塔池化(ASPP)模块对道路信息进行多尺度特征提取。最后，在解码网络部分加入特征对齐模块，调整由于上采样操作的不可学习性和上下采样的重复性导致的高低层特征之间存在的不准确的对应关系。本发明采用交叉熵和Dice系数组合的复合型损失函数对模型损失进行计算，缓解遥感道路数据集中存在的正负样本不平衡现象，提高了模型的道路提取结果。