[发明专利]一种基于GPU并行运算的以图搜图系统及方法

权利要求书

1.一种基于GPU并行运算的以图搜图系统，其特征在于，包括：

图片上传模块，用于将待搜索图片上传到系统中；

目标检测模块，用于对上传的图片进行检测和分割；

特征提取模块，用于对待搜索图片进行特征提取，并将图片抽象化成高维向量；

聚类索引模块，用于对图片库中未整理过的图片向量按照不同聚类类别分别进行聚类并建立一级索引，然后对初次聚类后得到的索引再次进行聚类并建立二级索引；

三级缓存模块，用于将所述的一级索引和二级索引分别预加载在内存和显存中，并将向量放在SSD固态硬盘上等待读取；

并行检索模块，用于依照所述的二级索引、一级索引、向量的顺序进行搜索，得到与待搜索图片最相似的搜索结果。

2.根据权利要求1所述的一种基于GPU并行运算的以图搜图系统，其特征在于，用户上传图片格式为jpg、png。

3.根据权利要求1所述的一种基于GPU并行运算的以图搜图系统，其特征在于，所述目标检测模块将图片中人、车、物分割作为候选搜索目标，另外，图片整体同样也作为候选搜索目标。

4.根据权利要求1所述的一种基于GPU并行运算的以图搜图系统，其特征在于，特征提取时将图片送入预训练好的模型来获取向量结果。

5.根据权利要求1所述的一种基于GPU并行运算的以图搜图系统，其特征在于，所述聚类类别包括图片时间、拍摄地点以及图片类别。

6.一种基于GPU并行运算的以图搜图方法，其特征在于，包括以下步骤：

S1、对图片库中未整理过的图片向量按照不同聚类类别分别进行聚类并建立一级索引，然后对初次聚类后得到的索引再次进行聚类并建立二级索引；

S2、将一级索引和二级索引分别预加载在内存和显存中，并将向量放在SSD固态硬盘上等待读取；

S3、将待搜索图片上传到系统中；

S4、对上传的图片进行检测和分割；

S5、对待搜索图片进行特征提取，并将待搜索图片抽象化成高维向量；

S6、依照二级索引、一级索引、向量的顺序进行搜索，得到与待搜索图片最相似的搜索结果。

7.根据权利要求6所述的一种基于GPU并行运算的以图搜图方法，其特征在于，用户上传图片格式为jpg、png。

8.根据权利要求6所述的一种基于GPU并行运算的以图搜图方法，其特征在于，S4步骤将图片中人、车、物分割作为候选搜索目标，另外，图片整体同样也作为候选搜索目标。

9.根据权利要求6所述的一种基于GPU并行运算的以图搜图方法，其特征在于，特征提取时将图片送入预训练好的模型来获取向量结果。

10.根据权利要求6所述的一种基于GPU并行运算的以图搜图方法，其特征在于，所述聚类类别包括图片时间、拍摄地点以及图片类别。