[发明专利]图像检索的方法及装置、图像表征网络的训练方法及装置

说明书

本说明书实施例提供了一种图像检索方法。该检索方法包括：获取待检索的第一图像，并将该第一图像输入训练好的图像表征网络，得到第一表征向量；再对该第一表征向量进行二值化处理，得到该第一图像的二值表征向量；另一方面，获取多个候选图像的多个二值表征向量；进一步，分别计算该多个二值表征向量与所述第一图像的二值表征向量之间的向量距离，并基于该向量距离，从上述多个候选图像中召回与上述第一图像相似的图像。本说明书实施例还提供一种图像表征网络的训练方法，此训练方法可以保证图像特征在二值前后的检索性能保持一致。

技术领域

本说明书一个或多个实施例涉及计算机技术领域，尤其涉及一种图像检索的方法及装置，以及一种图像表征网络的训练方法及装置。

背景技术

许多领域都涉及图像检索，例如，搜索引擎提供图片搜索功能，支持用户输入图片进行搜索，并将检索到的相同或相近的图片返回给用户。然而，随着检索库中图片量级的不断增加，图片检索的计算量和耗时随之增加，已有的图像检索技术难以满足实时性要求，导致用户体验受到影响。

因此，迫切需要一种方案，可以有效降低图片检索的时延，从而提高用户体验。

发明内容

本说明书实施例描述一种图像检索的方法及装置，以及一种图像表征网络的训练方法及装置，可以在保证图像召回准确率的同时，显著减少图像检索的耗时，从而降低用户等待时长，有效提升用户体验。

根据第一方面，提供了一种图像检索方法，包括：获取待检索的第一图像，并将该第一图像输入训练好的图像表征网络，得到第一表征向量；对所述第一表征向量进行二值化处理，得到所述第一图像的二值表征向量；获取多个候选图像的多个二值表征向量，并计算其中各个二值表征向量与所述第一图像的二值表征向量之间的向量距离；基于所述向量距离，从所述多个候选图像中召回与所述第一图像相似的图像。

在一个实施例中，在获取待检测的第一图像之前，所述方法还包括：将所述多个候选图像分别输入所述训练好的图像表征网络，得到对应的多个第二表征向量；对所述多个第二表征向量分别进行所述二值化处理，得到并存储所述多个候选图像的多个二值表征向量；其中，获取多个候选图像的多个二值表征向量，包括：读取存储的多个二值表征向量。

在一个实施例中，计算其中各个二值表征向量与所述第一图像的二值表征向量之间的向量距离，包括：对所述各个二值表征向量与所述第一图像的二值表征向量进行相同位置元素的比对；根据所述比对的结果，确定所述向量距离。

根据第二方面，提供了一种图像表征网络的训练方法，包括：获取训练图像及其对应的标注类别；将所述训练图像输入图像表征网络，得到训练表征向量；利用预设函数将所述训练表征向量中的各元素分别映射为逼近第一数值或第二数值的逼近值，得到逼近表征向量；分别计算多个权重向量与所述逼近表征向量之间的余弦相似度，其中多个权重向量对应多个图像类别；基于计算出的多个余弦相似度和所述标注类别，训练所述图像表征网络和多个权重向量。

在一个实施例中，在训练所述图像表征网络和多个权重向量之后，所述方法还包括：获取待检索的第一图像，并将该第一图像输入训练好的图像表征网络，得到第一表征向量；将所述第一表征向量中的各元素二值化处理为所述第一数值或第二数值，得到所述第一图像的二值表征向量；获取多个候选图像的多个二值表征向量，并计算其中各个二值表征向量与所述第一图像的二值表征向量之间的汉明距离；基于所述汉明距离，从所述多个候选图像中召回与所述第一图像相似的图像。

在一个实施例中，所述第一数值为-1，第二数值为1。

在一个具体的实施例中，所述预设函数为双曲正切函数。

在一个具体的实施例中，分别计算多个权重向量与所述逼近表征向量之间的余弦相似度，包括：计算所述逼近表征向量与其向量维数的算数平方根之间的商，得到单位表征向量；对所述多个权重向量分别进行归一化处理，得到多个单位权重向量；计算所述单位表征向量与各个单位权重向量之间的点积，得到对应的余弦相似度。