[发明专利]一种图像检索方法及装置

说明书

本发明公开了一种图像检索方法及装置。所述方法包括：确定待查询图像的特征向量；将所述待查询图像的特征向量依据转换规则转换为二进制码，所述转换规则是至少根据图像库中各存储图像确定的；将所述待查询图像的二进制码与所述图像库中各存储图像的二进制码进行比较，确定所述待查询图像的相似图像；针对每张相似图像，根据所述相似图像的特征向量与所述待查询图像的特征向量间的相似度，确定所述待查询图像的查询结果。

技术领域

本发明涉及图像技术领域，尤其涉及一种图像检索方法及装置。

背景技术

目前，由于现有的图库数量很大，在图像检索方面，现有的深度学习算法在单机对海量图像数据训练和检索时经常需要将数据进行内存和硬盘之间的交换，容易造成图像检索过程复杂且耗时长的问题。

另外，现有技术在进行图像检索时，通常是使用训练好的模型提取图像的特征，并使用相似度判断模型计算测试图像的特征和数据库中的多个样本图像的特征之间的相似度，并将相似度最大的样本图像的标签作为测试图像的标签。由于图库数据规模日益增大，很难获得足够的标签数据，进行模型的训练，导致识别效率很低，模型训练难度增加等问题。

发明内容

本发明实施例提供一种图像检索方法及装置，以解决现有技术中图像检索过程复杂且耗时长的技术问题。

本发明实施例提供一种图像检索方法，所述方法包括：

确定待查询图像的特征向量；

将所述待查询图像的特征向量依据转换规则转换为二进制码，所述转换规则是至少根据图像库中各存储图像确定的；

将所述待查询图像的二进制码与所述图像库中各存储图像的二进制码进行比较，确定所述待查询图像的相似图像；

针对每张相似图像，根据所述相似图像的特征向量与所述待查询图像的特征向量间的相似度，确定所述待查询图像的查询结果。

一种可能的实现方式，所述将所述待查询图像的特征向量依据转换规则转换为二进制码之前，还包括：

根据各参照图像确定K维空间，在所述各参照图像的特征向量在K维空间的各投影方向的方差相同且最大；所述各参照图像包括所述各存储图像和所述待查询图像；K为正整数；

在所述K维空间中，求解所述各参照图像的特征向量的K个球体散列映射函数；所述K个球体散列映射函数作为所述转换规则。

一种可能的实现方式，所述转换规则通过以下方式确定：

根据各参照图像确定K维空间，所述各参照图像的特征向量在所述K维空间的投影距离的方差最大且相同；所述各参照图像为所述各存储图像；

在所述K维空间中，求解所述各参照图像的特征向量的K个球体散列映射函数；所述K个球体散列映射函数作为所述转换规则。

一种可能的实现方式，所述根据各参照图像确定K维空间，包括：

确定将所述各参照图像的特征向量投影至K维空间的投影矩阵，以使所述各参照图像的特征向量在所述K维空间的各投影方向上的投影距离的方差最大；

根据正交矩阵旋转所述投影矩阵，以使所述各参照图像的特征向量在旋转后的K维空间的各投影方向上的投影距离的方差相同；

根据旋转后的投影矩阵，确定所述K维空间。

一种可能的实现方式，所述将所述待查询图像的二进制码与所述图像库中各存储图像的二进制码进行比较，确定所述待查询图像的相似图像，包括：

通过确定所述待查询图像的二进制码与所述各存储图像的二进制码的汉明距离，将汉明距离小于第一阈值的存储图像，确定为所述待查询图像的相似图像；