[发明专利]图像检索方法、装置、电子设备及计算机存储介质

说明书

本发明实施例公开了一种图像检索方法、装置、电子设备及计算机存储介质。该图像检索方法，包括：获取待检索图像的压缩向量；基于压缩向量和预设码本，分别确定压缩向量与预设码本中各个码字之间的距离；其中，预设码本为基于样本训练图像，利用梯度下降算法确定的码本；码字表征样本训练图像所属的类别信息；将各个距离中最小距离对应的码字，确定为目标码字；从目标码字对应的数据库中确定待检索图像的目标图像。根据本发明实施例，能够提高图像检索的精确度。

技术领域

本发明属于电子数据处理技术领域，尤其涉及一种图像检索方法、装置、电子设备及计算机存储介质。

背景技术

随着互联网的发展，对图像检索的要求也逐日提高。目前，常见的图像检索方法有如下几种：

穷尽搜索：对每一个需要查询的图像，都搜索整个数据集，并一一对比，最终得出搜索结果；乘积量化：对每一个需要查询的图像，先对其进行维度划分，假定其独立，根据划分之后的维度，在对应的维度里面进行穷尽搜索；集束量化(原始加法量化)：对每一个需要查询的图像，相对于乘积量化，取消了维度划分间独立性的要求，降低了约束条件要求，理论上效果更好。但是在集束搜索的过程中，耗费的时间和资源太多。

但是，这些图像检索方法在图像库的数据量提高后就不能短时间响应完成检索，且精度达不到应用的标准。

因此，如何提高图像检索的精确度是本领域技术人员亟需解决的技术问题。

发明内容

本发明实施例提供一种图像检索方法、装置、电子设备及计算机存储介质，能够提高图像检索的精确度。

第一方面，提供了一种图像检索方法，包括：

获取待检索图像的压缩向量；

基于压缩向量和预设码本，分别确定压缩向量与预设码本中各个码字之间的距离；其中，预设码本为基于样本训练图像，利用梯度下降算法确定的码本；码字表征样本训练图像所属的类别信息；

将各个距离中最小距离对应的码字，确定为目标码字；

从目标码字对应的数据库中确定待检索图像的目标图像。

可选地，基于压缩向量和预设码本，分别确定压缩向量与预设码本中各个码字之间的距离，包括：

获取预设码本对应的复合索引；

根据复合索引，确定预设码本中各个码字；

基于压缩向量和预设码本，分别确定压缩向量与各个码字之间的距离。

可选地，在获取预设码本对应的复合索引之前，方法还包括：

利用局部禁忌搜索算法，确定预设码本对应的复合索引。

可选地，在利用局部禁忌搜索算法，确定预设码本对应的复合索引之后，方法还包括：

存储预设码本和复合索引；

分别确定预设码本和复合索引的内存消耗信息。

可选地，方法还包括：

基于预设的目标函数，更新预设码本。

可选地，从目标码字对应的数据库中确定待检索图像的目标图像，包括：

基于待检索图像，穷尽搜索目标码字对应的数据库，确定目标图像。

可选地，在从目标码字对应的数据库中确定待检索图像的目标图像之后，方法还包括：

确定待检索图像对应的量化损失和/或召回率和/或准确率。

第二方面，提供了一种图像检索装置，包括：