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

说明书

本公开提供了一种图片检索方法及装置，其中，该方法包括：按照预设的多个尺寸，分别对第一图片和第二图片进行特征提取，获得所述第一图片对应的多个第一特征图和所述第二图片对应的多个第二特征图；计算位于任意两个空间位置上的所述第一特征图和所述第二特征图之间的相似度值，获得与目标尺寸组合对应的所述相似度值；根据与每个所述目标尺寸组合对应的所述相似度值，建立目标无向图；将所述目标无向图输入预先建立的目标图神经网络，根据所述目标图神经网络的输出结果，确定所述第二图片是否属于与所述第一图片匹配的目标图片。本公开结合预设的多个尺寸进行相似度分析，确定两张图片之间是否匹配，匹配精度更高，鲁棒性更强。

技术领域

本公开涉及图片处理领域，尤其涉及一种图片检索方法及装置。

背景技术

相关技术中，在将已有图片与图片库中的图片进行匹配搜索时，可以采用神经网络计算两张图片的全局相似度，从而在图片库中找到与已有图片匹配的图片。

但是，在计算两张图片的全局相似度时，图片中的背景干扰信息会对计算结果造成较大影响，例如图片的角度不同、图片的内容信息不同或遮挡等原因，会造成最终搜索的结果不准确。

发明内容

本公开提供了一种图片检索方法及装置。

根据本公开实施例的第一方面，提供一种图片检索方法，所述方法包括：按照预设的多个尺寸，分别对第一图片和第二图片进行特征提取，获得所述第一图片对应的多个第一特征图和所述第二图片对应的多个第二特征图；其中，所述第二图片是图片库中的任一图片；计算位于任意两个空间位置上的所述第一特征图和所述第二特征图之间的相似度值，获得与目标尺寸组合对应的所述相似度值；其中，所述目标尺寸组合包括所述预设的多个尺寸中的第一尺寸和第二尺寸，所述第一特征图对应所述第一尺寸，所述第二特征图对应所述第二尺寸；根据与每个所述目标尺寸组合对应的所述相似度值，建立目标无向图；将所述目标无向图输入预先建立的目标图神经网络，根据所述目标图神经网络的输出结果，确定所述第二图片是否属于与所述第一图片匹配的目标图片。上述实施例中，可以按照预设的多个尺寸，分别对第一图片和图片库中的第二图片进行特征提取，获得第一图片对应的多个第一特征图和第二图片对应的多个第二特征图，计算位于任意两个空间位置上的第一特征图和第二特征图之间的相似度值，获得与目标尺寸组合对应的相似度值。根据与每个目标尺寸组合对应的相似度值，建立目标无向图。将目标无向图输入预先建立的目标图神经网络，可以确定出第二图片是否属于与第一图片匹配的目标图片。通过上述过程，不再局限于两张图片的整体尺寸去进行全局相似度分析，而是结合预设的多个尺寸进行相似度分析，根据对应第一尺寸的第一图片的第一特征图和对应第二尺寸的第二图片的第二特征图位于任意两个空间位置的局部相似度值，来确定两张图片之间是否匹配，匹配精度更高，鲁棒性更强。

在一些可选实施例中，所述预设的多个尺寸包括第三尺寸和至少一个第四尺寸，所述第三尺寸是包括所述第一图片中的所有像素点在内的尺寸，所述第四尺寸小于所述第三尺寸。上述实施例中，预设的多个尺寸包括了第三尺寸和至少一个第四尺寸，第三尺寸是第一图片的整体尺寸，第四尺寸可以小于第三尺寸，从而在计算第一图片和第二图片的相似度时，不再局限于两张图片的整体相似度，而是考虑到了不同尺寸下的图片之间的相似度，可以提高匹配结果的精度，鲁棒性更好。