[发明专利]图像特征搜索方法、装置、电子设备及存储介质

说明书

本发明实施例提供一种图像特征搜索方法、装置、电子设备及存储介质，所述方法包括：获取待搜索图像特征以及待选图像特征；将待搜索图像特征以及待选图像特征进行自相关归一化；对自相关归一化后的所述待搜索图像特征在所述待选图像特征中进行暴力搜索，得到暴力搜索结果；基于所述暴力搜索结果，选取目标结果进行输出。通过对待搜索图像特征以及待选图像特征进行自相关归一化，使得到计算相似度时，无需对待搜索图像特征以及待选图像特征进行自相关运算，只需要执行点积运算即可，一次相似度计算减少了两次自相关运算，提高了相似度计算的速度，从而提高了图像搜索的速度。

技术领域

本发明涉及图像搜索技术领域，尤其涉及一种图像特征搜索方法、装置、电子设备及存储介质。

背景技术

图像搜索是人工智能领域中常用的技术之一，例如：人脸搜索、物体搜索、车辆搜索等。图像搜索一般是将提取出的特征值与数据库存储的特征值进行逐一进行比对，比对出相似度最高的特征值所对应的图像作为搜索结果，随着数据规模的增大，数据生命周期的增加，又因为图像搜索需要在海量的特征值中逐一进行比对，使得搜索的计算量也会增加，从而导致图像搜索的速度变慢。因此，现有的图像搜索存在搜索速度慢的问题。

发明内容

本发明实施例提供一种图像特征搜索方法、装置、电子设备及存储介质，能够提高图像搜索速度。

第一方面，本发明实施例提供一种图像特征搜索方法，包括：

获取待搜索图像特征以及待选图像特征；

将待搜索图像特征以及待选图像特征进行自相关归一化；

对自相关归一化后的所述待搜索图像特征在所述待选图像特征中进行暴力搜索，得到暴力搜索结果；

基于所述暴力搜索结果，选取目标结果进行输出。

可选的，所述对自相关归一化后的所述待搜索图像特征在所述待选图像特征中进行暴力搜索，得到暴力搜索结果，包括：

计算自相关归一化后的所述待搜索图像特征以及待选图像特征的余弦相似度；

选取余弦相似度满足预设阈值的待选图像特征，作为所述暴力搜索结果。

可选的，所述计算所述待搜索图像特征以及待选图像特征的余弦相似度，包括：

通过循环展开完成预设数量的余弦相似度的计算，所述预设数量与所述循环展开的展开度相同。

可选的，所述循环展开基于avx2指令集进行配置。

可选的，所述获取待选图像特征，包括：

提取待搜索图像特征的待搜索短特征值；

将所述待搜索短特征值与图像数据库中的短特征值进行相似度TOPK计算，得到预设数量的短特征值结果；

根据所述短特征值结果索引对应的图像特征，作为待选图像特征；

其中，所述待搜索短特征值与所述图像数据库中的短特征值通过同一特征提取引擎进行提取，所述图像数据库中的短特征值与对应的图像特征存在映射关系。

可选的，所述方法还包括：

获取搜索条件，根据所述搜索条件确定图像数据库中的搜索区域，所述搜索区域包括与所述搜索条件对应的短特征值；

所述将所述待搜索短特征值与图像数据库中的短特征值进行相似度TOPK计算，得到预设数量的短特征值结果，包括：

将所述待搜索短特征值与所述搜索区域中的短特征值进行相似度TOPK计算，得到预设数量的短特征值结果。

可选的，所述搜索条件包括目标属性值，所述待选图像包括属性值，所述方法还包括：