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

作者：张向荣; 焦李成; 韩骁; 唐旭; 白静; 冯婕; 侯彪; 马文萍

来源：2018-06-08, 中国, ZL201810590208.1.

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