[发明专利]一种检索方法和装置

说明书

本申请公开了一种检索方法和装置，方法包括：将第一存储介质中的第一距离表和第一索引集合存储至第二存储介质；第一存储介质中存储有至少两个特征向量的索引以及基于待检索特征向量确定的多个距离表；特征向量的索引包括一级索引和二级索引，距离表与一级索引一一对应；第一索引集合是至少两个特征向量中的第一类特征向量的二级索引构成的集合，第一类特征向量的一级索引与第一距离表对应的一级索引相同；根据第二存储介质中存储的第一距离表和第一索引集合中的二级索引，获取第一类特征向量与待检索特征向量之间的距离；根据第一类特征向量与待检索特征向量之间的距离，在至少两个特征向量中确定与待检索向量匹配的特征向量。

技术领域

本申请涉及检索技术领域，尤其涉及一种检索方法和装置。

背景技术

“以图搜图”是公安、安防系统中被广泛用于查找目标的检索方法。以图搜图是在多个预设图像中确定与待检索图像匹配的图像的技术。具体地：在数据库中预设多个图像对应的多个特征向量(或称为长特征向量)，预设图像与特征向量一一对应，预设图像对应的特征向量可以是采用深度学习算法(如人工智能(artificial intelligence，AI)算法)提取该预设图像的特征信息，并基于所提取的特征信息构建的向量；当需要对待检索图像进行检索时，分别计算待检索图像的特征向量与每个预设图像对应的特征向量之间的距离；然后将与待检索图像的特征向量的距离最小的特征向量对应的预设图像作为待检索图像匹配的图像。

由于数据库中预存的预设图像的数量非常大，一个数据库中往往预存几百万、几千万、甚至几亿个图像，上述检索方法在实际应用中耗时长，检索效率低，如何提供一种高效的检索方法成为亟待解决的技术问题。

发明内容

本申请提供了一种检索方法和装置，以解决传统技术中检索过程耗时长，效率低的问题。

第一方面，本申请提供了一种检索方法，该方法可以包括：将第一存储介质中的第一距离表和第一索引集合存储至第二存储介质；其中，第一存储介质中存储有至少两个特征向量的索引以及基于待检索特征向量确定的多个距离表；特征向量的索引包括一级索引和二级索引，距离表与一级索引一一对应；第一距离表是该多个距离表中的任一距离表；第一索引集合是至少两个特征向量中的第一类特征向量的二级索引构成的集合，第一类特征向量的一级索引与第一距离表对应的一级索引相同；第一存储介质和第二存储介质归属于同一个服务器；根据第二存储介质中存储的第一距离表和第一索引集合中的二级索引，获取第一类特征向量与待检索特征向量之间的距离；根据第一类特征向量与待检索特征向量之间的距离，在至少两个特征向量中确定与待检索向量匹配的特征向量。

其中，该技术方案的执行主体可以是服务器，具体可以是服务器中的处理器。一级索引，可以用于指示该特征向量所属的簇的簇号。一级索引具体可以是K重一级索引中的其中一重一级索引，其中，K≥1，K是整数。例如，K＝1或K＝2或K＝4。二级索引，可以用于指示该特征向量经该特征向量所属的簇对应的码书处理后得到的量化索引。索引集合，可以是同一个簇中的部分或全部特征向量的二级索引构成的集合。距离表是待检索特征向量中的多个子特征向量与码书中相应中心点之间的距离值的集合。码书与簇一一对应。关于一级索引、第二级索引、距离表和码书等术语的具体说明可以参考下文具体实施方式部分。

可选地，第一存储介质的读写性能高于第二存储介质的存储性能。例如，第一存储介质包括内存(memory)或显存(GPU memory)，第二存储介质包括高速缓存(cache)。当第一存储介质是内存时，处理器可以是中央处理器(central processing unit，CPU)等。当第一存储介质是显存时，处理器可以是图形处理器(graphics processing unit，GPU)等。