原文传递
基于生成对抗网络的遥感图像道路提取方法
| 专利名称： | 基于生成对抗网络的遥感图像道路提取方法 |
| 摘要： | 本发明提出一种基于生成对抗网络的遥感图像道路提取方法，解决了遥感图像道路提取准确率低，道路连续性不佳的问题。实现步骤为：在现有遥感数据库中选取样本集并划分测试样本和训练样本；设置生成对抗网络结构，包括生成网络和判别网络；设定用于道路提取的生成对抗网络失函数，包括传统生成对抗网络损失和针对于道路提取的欧式距离损失；用自适应矩估计法训练生成对抗网络得到网络参数；训练好的模型对测试样本进行道路提取；输出有路网和背景信息的遥感图像。本发明提高了道路提取准确性，提取的路网更加完整、道路连续性更好，准确率、查全率、查准率与F1‑score都有显著提升，用于遥感图像的道路提取。 |
| 专利类型： | 发明专利 |
| 国家地区组织代码： | 陕西;61 |
| 申请人： | 西安电子科技大学 |
| 发明人： | 张向荣;焦李成;韩骁;唐旭;白静;冯婕;侯彪;马文萍 |
| 专利状态： | 有效 |
| 发布日期： | 2019-01-01T00:00:00+0800 |
| 申请号： | CN201810590208.1 |
| 公开号： | CN108830209A |
| 代理机构： | 陕西电子工业专利中心 61205 |
| 代理人： | 程晓霞;王品华 |
| 分类号： | G06K9/00(2006.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I;G;G06;G06K;G06N;G06K9;G06N3;G06K9/00;G06N3/04;G06N3/08 |
| 申请人地址： | 710071 陕西省西安市雁塔区太白南路2号 |
| 主权项： | 1.一种基于生成对抗网络的遥感图像道路提取方法，其特征在于，包括步骤如下：(1)遥感图像划分为训练样本和测试样本：对输入大小为N×N的遥感图像以及遥感图像对应的二值类标图像进行划分，其中多半作为训练样本，其余作为测试样本；(2)设置用于遥感图像道路提取的生成对抗网络结构：(2a)生成对抗网络中的生成网络结构设置为全卷积神经网络；(2b)生成对抗网络中的判别网络结构设置为二分类卷积神经网络；(2c)生成网络与判别网络共同构成生成对抗网络；(3)设定用于道路提取的生成对抗网络的损失函数Loss：将包含道路信息的正样本惩罚和负样本惩罚的生成对抗网络损失与提取结果与真实道路信息之间的欧式距离损失分别加权后进行求和，其和值设定为用于道路提取的生成对抗网络的损失函数Loss；(4)对生成对抗网络进行训练：采用自适应矩估计优化算法，通过训练样本和用于道路提取的生成对抗网络的损失函数，对生成对抗网络进行训练，得到包含模型参数的生成对抗网络；(4a)采用梯度上升的自适应矩估计优化算法，通过总体的损失函数来训练判别网络；(4b)采用梯度下降的自适应矩估计优化算法，通过总体的损失函数来训练生成网络；(4c)反复进行(4a)、(4b)两个步骤，轮换对判别网络和生成网络进行训练，直到模型参数收敛，得到训练好的模型；(5)用训练好的模型对测试样本进行道路提取：将测试样本中的遥感图像输入已经训练好的生成网络中，得到只包含道路与背景的二值类标图，即为所需提取过道路的遥感图像；(6)输出提取过道路的遥感图像。 |
| 所属类别： | 发明专利 |
检索历史
应用推荐
