[发明专利]图像搜索方法、系统及电子装置

权利要求书

1.一种图像搜索方法，其特征在于，所述方法包括：

获取待搜索图像；

提取所述待搜索图像的第一特征，基于所述第一特征提取所述待搜索图像的第二特征；所述第二特征的数据量小于第一特征的数据量；

从预先建立的图片特征库中搜索与所述待搜索图像的第二特征相匹配的图像，得到第一搜索结果；

从所述第一搜索结果中搜索与所述待搜索图像的第一特征相匹配的图像，得到第二搜索结果；

根据所述第二搜索结果确定所述搜索图像的最终搜索结果。

2.根据权利要求1所述的方法，其特征在于，所述图片特征库通过下述方式建立：

获取指定数量的样本图像；

对所述样本图像进行特征提取，得到所述样本图像的第一特征，将所述样本图像的标识信息和所述样本图像对应的第一特征保存至第一特征库中；

将所述第一特征输入至预先完成训练的深度学习模型中，输出所述样本图像的第二特征，将所述样本图像的标识信息和所述样本图像对应的第二特征保存至第二特征库中；

将所述第一特征库、所述第二特征库确定为图片特征库。

3.根据权利要求2所述的方法，其特征在于，基于所述第一特征提取所述待搜索图像的第二特征的步骤，包括：

将所述待搜索图像的第一特征输入至预先完成训练的深度学习模型，输出所述待搜索图像的第二特征。

4.根据权利要求2所述的方法，其特征在于，从预先建立的图片特征库中搜索与所述待搜索图像的第二特征相匹配的图像，得到第一搜索结果的步骤，包括：

将所述待搜索图像的第二特征与所述图片特征库中的所述第二特征库进行相似度计算，得到所述待搜索图像与所述图片特征库中每个图片的第一相似度计算结果；

将所述第一相似度计算结果中，相似度高于第一预设阈值的样本图像作为第一搜索结果。

5.根据权利要求4所述的方法，其特征在于，从预先建立的图片特征库中搜索与所述待搜索图像的第二特征相匹配的图像，得到第一搜索结果的步骤，包括：

将所述待搜索图像的第二特征、以及用户输入的搜索条件复制到GPU中；

通过GPU从预先建立的图片特征库中搜索与所述待搜索图像的第二特征相匹配的图像，得到第一搜索结果；

将所述第一搜索结果保存在内存中。

6.根据权利要求2所述的方法，其特征在于，所述从第一搜索结果中搜索与所述待搜索图像的第一特征相匹配的图像，得到第二搜索结果的步骤，包括：

从所述图片特征库的所述第一特征库中，提取所述第一搜索结果中样本图像的第一特征；

将所述待搜索图像的第一特征，分别与提取出的每个样本图像的第一特征进行相似度计算，得到所述待搜索图像的第一特征与每个提取出的每个样本图像的第一特征的第二相似度计算结果；

将所述第二相似度计算结果中，相似度高于第二预设阈值的样本图像作为第二搜索结果。

7.根据权利要求6所述的方法，其特征在于，所述从第一搜索结果中搜索与所述待搜索图像的第一特征相匹配的图像，得到第二搜索结果的步骤，包括：

将所述待搜索图像的第一特征、以及用户输入的搜索条件复制到CPU中；

通过CPU从所述待搜索图像的第一特征与每个提取出的每个样本图像中，搜索与所述待搜索图像的第一特征相匹配的图像，得到第二搜索结果。

8.根据权利要求1所述的方法，其特征在于，根据所述第二搜索结果确定所述待搜索图像的最终搜索结果的步骤，包括：

按照相似度由高到低的顺序，对所述第二搜索结果中的样本图像进行排序；

从排列在第一个的样本图像开始，依次提取指定数量的样本图像；

将提取出的所述样本图像、以及所述样本图像的属性信息作为最终的搜索结果。