[发明专利]获取图像检索模型的方法、图像检索方法、装置和设备

说明书

本申请公开了一种获取图像检索模型的方法、图像检索方法、装置和设备，属于计算机技术领域。方法包括：获取样本图像和初始图像检索模型，初始图像检索模型包括特征模块、特征量化模块和量化码本模块，特征模块用于获取样本图像对应的特征向量，特征量化模块用于获取特征向量对应的量化向量，量化码本模块用于提供量化码本，量化码本随着初始图像检索模型的更新进行更新；调用特征模块获取样本图像对应的样本特征向量；调用特征量化模块获取样本特征向量对应的样本量化向量；基于样本量化向量和量化码本确定目标损失值；基于目标损失值对初始图像检索模型进行更新，得到目标图像检索模型。该方法得到的目标图像检索模型的检索准确性较高。

技术领域

本申请实施例涉及计算机技术领域，特别涉及一种获取图像检索模型的方法、图像检索方法、装置和设备。

背景技术

随着计算机技术的不断发展，越来越多的领域涉及图像检索，而图像检索的方式也越来越多，采用图像检索模型进行图像检索便是其中的一种。

在采用图像检索模型进行图像检索时，图像检索模型的性能决定了检索结果的准确性，因此，如何获取性能更好的图像检索模型，是提高图像检索结果的准确性的关键。

发明内容

本申请实施例提供了一种获取图像检索模型的方法、图像检索方法、装置和设备，可用于提高图像检索的准确性。所述技术方案如下：

一方面，本申请实施例提供了一种获取图像检索模型的方法，所述方法包括：

获取样本图像和初始图像检索模型，所述初始图像检索模型包括特征模块、特征量化模块和量化码本模块，所述特征模块用于获取所述样本图像对应的特征向量，所述特征量化模块用于获取所述特征向量对应的量化向量，所述量化码本模块用于提供量化码本，所述量化码本随着所述初始图像检索模型的更新进行更新；

调用所述特征模块获取所述样本图像对应的样本特征向量；

调用所述特征量化模块获取所述样本特征向量对应的样本量化向量；

基于所述样本量化向量和所述量化码本确定目标损失值；

基于所述目标损失值对所述初始图像检索模型进行更新，得到目标图像检索模型。

另一方面，本申请实施例提供了一种图像检索方法，所述方法包括：

获取待检索的目标图像和目标图像检索模型，所述目标图像检索模型如上述获取图像检索模型的方法得到；

将所述目标图像输入至所述目标图像检索模型，得到所述目标图像对应的量化向量；

基于所述目标图像对应的量化向量与所述目标图像检索模型中的量化码本获取图像检索结果。

另一方面，本申请实施例提供了一种获取图像检索模型的装置，所述装置包括：

获取模块，用于获取样本图像和初始图像检索模型，所述初始图像检索模型包括特征模块、特征量化模块和量化码本模块，所述特征模块用于获取所述样本图像对应的特征向量，所述特征量化模块用于获取所述特征向量对应的量化向量，所述量化码本模块用于提供量化码本，所述量化码本随着所述初始图像检索模型的更新进行更新；

所述获取模块，用于调用所述特征模块获取所述样本图像对应的样本特征向量；

所述获取模块，用于调用所述特征量化模块获取所述样本特征向量对应的样本量化向量；

确定模块，用于基于所述样本量化向量和所述量化码本确定目标损失值；

更新模块，用于基于所述目标损失值对所述初始图像检索模型进行更新，得到目标图像检索模型。