[发明专利]基于强化学习的遥感影像的语义标注方法

摘要

本发明提供一种基于强化学习的遥感影像语义标注方法，包括如下几个步骤：数据获取；数据预处理；切割影像数据；制作样本集标注；构建强化学习网络模型；设置训练参数；选取训练集、验证集；使用训练集训练视觉语义嵌入网络；使用训练集训练value网络；使用训练集训练policy网络；使用训练集联合训练policy、value网络；使用验证集进一步优化网络参数；验证模型效果。本发明的目的就是将计算机视觉领域与遥感领域相结合，以强化学习的理念为基础对传统的VGG接GRU模型进行改进，进而解决传统方法的大样本量、训练周期长的问题，同时利用强化学习中实时反馈的机制可以减少目前遥感中难以解决的“同物异谱”、“异物同谱”问题对精度的影响。

主权项

1.基于强化学习的遥感影像的语义标注方法，其特征在于，包括如下步骤：步骤S1，数据获取：获取研究地段的高分辨率遥感影像；步骤S2，数据预处理：对已经获取的高分辨率遥感数据进行预处理，包括几何校正、大气校正、裁剪处理；步骤S3，切割影像数据：通过对原始影像的分析，确定影像切割的尺度，即单个样本数据的大小，用掩膜提取的方法将数据从原始影像中切割出来，将切割出来的样本以ID加影像格式后缀名的形式保存，所有切割完的样本成为一个样本集；步骤S4，制作样本集的标注：按照样本ID添加标注，即为描述样本内容的句子；步骤S5，构建强化学习网络模型：构建的网络模型包括policy网络、value网络、嵌入网络三个网络；其中policy网络用于在每个时刻输出下一时刻出词的候选列表以及列表中各词的概率；嵌入网络用来训练样本影像对应样本标注语句的reward；value网络用于输出一个候选列表以及列表中各词的value值(表示词汇与图像的关联性)；步骤S6，设置训练参数：设置VGG16网络的batch_size、学习率、迭代次数、词汇表大小、保存路径；对于GRU网络而言，参数初始化采用随机初始化方法；步骤S7，选取训练集、验证集：将样本集按比例分为训练集和验证集；步骤S8，使用训练集训练嵌入网络：输入样本影像及对应的标注语句，训练句子与图像的reward值，当嵌入网络训练完成后输入标注语句与样本影像得到的reward值最高(即生成句子越接近标注语句，reward值越大)，此时嵌入网络可以最大程度上判断句子与图像的相似度，训练完成嵌入网络加以保存；步骤S9，使用训练集训练value网络：将影像样本及对应的标注语句输入value网络，最后输出value值，与嵌入网络输出reward值的误差进行参数优化，训练完成value网络加以保存；步骤S10，使用训练集训练policy网络：将样本影像及其对应的标注语句输入policy网络，通过优化输出值与标签值的loss函数优化参数，训练完成policy网络加以保存；步骤S11，使用训练集联合训练policy、value网络：导入步骤S9、S10训练完成的value与policy网络参数，输入样本影像，联合policy网络输出的概率值以及value网络输出的value值，将整个出词的过程视为部分可观测马尔可夫决策，使用强化学习的方式联合训练policy和value网络，使用策略梯度函数调整两个网络的参数，更新并保存最终训练完的模型；步骤S12，使用验证集进一步优化网络参数：读取训练完成的policy、value网络，输入样本数据，调整输出结果与标签数据之间的误差，并更新网络参数进行下次训练，迭代至误差曲线稳定时模型训练完成，保存训练模型。