[发明专利]一种图像检索方法、装置和存储介质

说明书

本发明实施例公开了一种图像检索方法、装置和存储介质；本发明实施例在接收用户发送的图像检索请求后，该图像检索请求携带数据库标识和需要检索的图像特征，可以获取该数据库标识对应的目标数据库，该目标数据库包括多个子库，接着，基于子库中的数据为每个子库配置预设大小的显存空间，然后，将子库中的数据存储至该显存空间中，得到每个子库对应的存储块，再根据该图像特征并行对所有存储块中的数据进行检索，得到每个存储块的初始检索结果，再然后，对所有存储块中的初始检索结果进行整合，得到检索结果；该方案可以提高显存的利用率。

技术领域

本发明涉及通信技术领域，具体涉及一种图像检索方法、装置和存储介质。

背景技术

随着计算机网络技术和数字媒体技术的快速发展，人类进入了一个信息高度发达的互联网时代。每天网络上都会产生海量的数据信息，如何从包含海量的数据库中快速、准确地检索出用户所需的数据信息，无论是在时间上还是空间上都是一个极大的考验。随着科技的发展，显卡中图形处理器(Graphics Processing Unit，GPU)的处理能力也在不断增强，GPU越来越多地被应用于计算密集型的数据运算处理中，比如，图像检索。但是，现有的图像检索技术，一般是为一个静态检索库按照需求上限预分配显卡中显存，显存利用率低。

发明内容

本发明实施例提供一种图像检索方法、装置和存储介质，可以提高显存的利用率。

本发明实施例提供一种图像检索方法，包括：

接收用户发送的图像检索请求，所述图像检索请求携带数据库标识和需要检索的图像特征；

获取所述数据库标识对应的目标数据库，所述目标数据库包括多个子库；

基于子库中的数据为每个子库配置预设大小的显存空间；

将子库中的数据存储至所述显存空间中，得到每个子库对应的存储块；

根据所述图像特征并行对所有存储块中的数据进行检索，得到每个存储块的初始检索结果；

对所有存储块中的初始检索结果进行整合，得到检索结果。

相应的，本发明实施例还提供一种图像检索装置，包括：

接收单元，用于接收用户发送的图像检索请求，所述图像检索请求携带数据库标识和需要检索的图像特征；

获取单元，用于获取所述数据库标识对应的目标数据库，所述目标数据库包括多个子库；

配置单元，用于基于子库中的数据为每个子库配置预设大小的显存空间，并将子库中的数据存储至所述显存空间中，得到每个子库对应的存储块；

检索单元，用于根据所述图像特征并行对所有存储块中的数据进行检索，得到每个存储块的初始检索结果；

整合单元，用于对所有存储块中的初始检索结果进行整合，得到检索结果。

可选的，在一些实施例中，所述子库中的数据包括多个检索特征，所述检索单元可以包括读取子单元、计算子单元和第一筛选子单元，如下：

所述读取子单元，用于利用图形处理器读取每个存储块中的检索特征；

所述计算子单元，用于并行计算读取到的检索特征与所述图像特征的相似度，得到每个存储块对应的相似度集；

所述第一筛选子单元，用于从每个存储块对应的相似度集中筛选出与所述图像特征相似度最高的检索特征，得到每个存储块的初始检索结果。

可选的，在一些实施例中，所述计算子单元，具体可以用于将所述图像特征和每个存储块中的每个检索特征进行向量化，得到图像特征向量和多个检索向量；并行计算所述图像特征向量和所述多个检索向量的距离，得到多个距离值；