基于生成对抗网络的遥感图像道路提取方法

摘要

本发明提出一种基于生成对抗网络的遥感图像道路提取方法，解决了遥感图像道路提取准确率低，道路连续性不佳的问题。实现步骤为：在现有遥感数据库中选取样本集并划分测试样本和训练样本；设置生成对抗网络结构，包括生成网络和判别网络；设定用于道路提取的生成对抗网络失函数，包括传统生成对抗网络损失和针对于道路提取的欧式距离损失；用自适应矩估计法训练生成对抗网络得到网络参数；训练好的模型对测试样本进行道路提取；输出有路网和背景信息的遥感图像。本发明提高了道路提取准确性，提取的路网更加完整、道路连续性更好，准确率、查全率、查准率与F1‑score都有显著提升，用于遥感图像的道路提取。