[发明专利]相似图搜索方法及装置

说明书

本公开揭示了一种相似图搜索方法及装置，属于图像匹配领域。所述相似图搜索方法包括：在为第一图像搜索相似图像时，将第一图像输入至图像分类模型Alexnet中，控制Alexnet输出第一图像的预定维特征；从预定图库中获取第二图像，将第二图像输入至Alexnet中，控制Alexnet输出第二图像的预定维特征；计算第一图像的预定维特征和第二图像的预定维特征之间的匹配度；当计算得到的匹配度小于预定阈值时，判定第二图像为第一图像的相似图像。解决了较平滑的图像无法成功提取出局部特征点，而导致无法实现相似图搜索的问题；达到了成功提取出较平滑图像局部特征点，提高相似图搜索率的效果。

技术领域

本公开涉及图像匹配领域，特别涉及一种相似图搜索方法及装置。

背景技术

由于相似图搜索技术可以从海量的图像中自动识别出相似的图像，因此使很多基于相似图像搜索的应用成为可能。

相关技术中在进行相似图像搜索时，首先，对需要搜索相似图的图像，提取该图像中的局部特征点，对提取出的局部特征点进行特征描述，得到特征描述文档，将这些特征描述文档组成文档倒排索引树；然后，从图库中获取一张图像，基于类似方式，获取该图像的文档倒排索引树，在匹配两张图像是否为相似图像时，根据这两张图像的文档倒排索引树，筛选出匹配的特征文档，根据筛选出的匹配的特征文档来判定这两张图像是否相似。

由于很多较平滑的图像无法成功提取出局部特征点，因此无法实现这些图像的相似图搜索。

发明内容

为了解决相关技术中由于很多较平滑的图像无法成功提取出局部特征点，因此无法实现这些图像的相似图搜索的问题，本公开提供一种相似图搜索方法及装置。所述技术方案如下：

根据本公开实施例的第一方面，提供一种相似图搜索方法，所述方法包括：

在为第一图像搜索相似图像时，将所述第一图像输入至图像分类模型Alexnet中，控制所述Alexnet输出所述第一图像的预定维特征；

从预定图库中获取第二图像，将所述第二图像输入至所述Alexnet中，控制所述Alexnet输出所述第二图像的预定维特征；

计算所述第一图像的预定维特征和所述第二图像的预定维特征之间的匹配度；

当计算得到的匹配度小于预定阈值时，判定所述第二图像为所述第一图像的相似图像。

本公开提供的上述相似图搜索方法能够达到的有益效果为：通过将第一图像和预定图库中获取的第二图像均输入至Alexnet中，控制Alexnet输出该第一图像的预定维特征和第二图像的预定维特征，当该第一图像的预定维特征和第二图像的预定维特征之间的匹配度小于预定阈值时，则判定该第二图像为该第一图像的相似图像；由于图像在输入Alexnet后，Alexnet所输出的该图像的预定维特征针对该图像具有较强的特征代表性，若两张图像的预定维特征的匹配度相近，通常可以表明这两张图像相似，且Alexnet模型针对平滑的图像也可以成功提取出预定维特征，因此解决了较平滑的图像无法成功提取出局部特征点，而导致无法实现相似图搜索的问题；达到了成功提取出较平滑图像局部特征点，提高相似图搜索率的效果。

可选的，所述将所述第一图像输入至图像分类模型Alexnet中，控制所述Alexnet输出所述第一图像的预定维特征，包括：

去除所述Alexnet中的最后一层分类器；将所述第一图像输入至去除最后一层分类器的所述Alexnet中，控制所述Alexnet输出所述第一图像的所述预定维特征；或者，

将所述第一图像输入至图像分类模型Alexnet中，向所述Alexnet下发特征输出指令，所述特征输出指令用于指示所述Alexnet在所述第一图像输入后，在将所述第一图像的预定维特征输入至最后一层分类器前，将得到的所述预定维特征输出。