一种基于多头自注意力机制的高分辨率遥感影像中道路提取方法、系统、设备及介质
摘要:
一种基于多头自注意力机制的高分辨率遥感影像中道路提取方法、系统、设备及介质,方法包括:对从卫星获取的高分辨率遥感影像进行预处理,得到高分辨率遥感影像数据集De,将高分辨率遥感影像数据集De按比例划分为训练集验证集和测试集设计搭建基于深度学习的高分辨率遥感影像道路提取网络;计算网络损失;设置网络训练参数;网络训练;网络测试;结果评估;系统、设备及介质,用于实现一种基于多头自注意力机制的高分辨率遥感影像中道路提取方法;本发明利用Canny边缘算子对道路遥感影像进行预处理,获得遥感影像中道路的边缘特征;基于U‑Net模型,融合多头自注意力机制模块,解决了现有技术参数量大,耗费时间多,道路提取精度低的问题。
0/0