[发明专利]一种遥感图像检索的方法和装置

说明书

本发明实施例提供了一种遥感图像检索的方法和装置，所述方法包括：获取遥感图像数据；将所述遥感图像数据划分为训练集、验证集，以及测试集；搭建用于对遥感图像数据检索的目标网络模型；采用所述训练集和所述验证集，对所述目标网络模型进行模型训练，得到在所述验证集上准确率最高的模型权重文件；根据所述模型权重文件，对所述目标网络模型进行调整，并采用所述调整后的目标网络模型对所述测试集进行检索，得到检索结果。通过本发明实施例，实现了对遥感图像检索，提升了检索的准确性。

技术领域

本发明涉及图像识别领域，特别是涉及遥感图像检索的方法和装置。

背景技术

随着遥感技术的快速发展,遥感图像数据库呈爆炸式增长,为了高效管理遥感图像数据库,基于内容的图像检索(CBIR)系统成为国内外研究的热点。

CBIR有两个主要步骤:图像的特征提取和图像的相似性匹配.早期的CBIR根据图像的纹理、颜色、形状等视觉特征进行提取,例如尺度不变特征转换(SIFT)算法和方向梯度直方图(HOG)算法,但这种低层的全局特征易受视角、光照、遮挡等影响。

发明内容

鉴于上述问题，提出了以便提供克服上述问题或者至少部分地解决上述问题的一种遥感图像检索的方法和装置，包括：

一种遥感图像检索的方法，所述方法包括：

获取遥感图像数据；

将所述遥感图像数据划分为训练集、验证集，以及测试集；

搭建用于对遥感图像数据检索的目标网络模型；

采用所述训练集和所述验证集，对所述目标网络模型进行模型训练，得到在所述验证集上准确率最高的模型权重文件；

根据所述模型权重文件，对所述目标网络模型进行调整，并采用所述调整后的目标网络模型对所述测试集进行检索，得到检索结果。

可选地，所述搭建用于对遥感图像数据检索的目标网络模型，包括：

构造空间注意力机制模块；

对所述空间注意力机制模块进行仿射变换，以搭建用于对遥感图像数据检索的目标网络模型。

可选地，所述对所述空间注意力机制模块进行仿射变换，包括：

确定仿射变换参数；

根据所述仿射变换参数，对所述空间注意力机制模块进行仿射变换。

可选地，在所述将所述遥感图像数据划分为训练集、验证集，以及测试集之前，还包括：

对所述遥感图像数据进行预处理和数据增广。

可选地，所述数据增广的方式包括但不限于：

对所述遥感图像数据进行翻转、对所述遥感图像数据进行平移、对所述遥感图像数据进行缩放、调整所述遥感图像数据的RGB各通道权重、对所述遥感图像数据进行旋转。

一种遥感图像检索的装置，所述装置包括：

遥感图像数据获取模块，用于获取遥感图像数据；

遥感图像数据划分模块，用于将所述遥感图像数据划分为训练集、验证集，以及测试集；

目标网络模型搭建模块，用于搭建用于对遥感图像数据检索的目标网络模型；

模型权重文件得到模块，用于采用所述训练集和所述验证集，对所述目标网络模型进行模型训练，得到在所述验证集上准确率最高的模型权重文件；

检索结果得到模块，用于根据所述模型权重文件，对所述目标网络模型进行调整，并采用所述调整后的目标网络模型对所述测试集进行检索，得到检索结果。