[发明专利]图像搜索方法、装置、电子设备及计算机可读存储介质

说明书

本公开涉及图像处理技术领域，提供了一种图像搜索方法、装置、电子设备及计算机可读存储介质。该方法包括：获取待搜索图像；将待搜索图像输入特征提取模型，得到待搜索图像的图像特征和质量分数；其中，特征提取模型是基于样本图像质量分数进行自适应间隔训练确定，样本图像质量分数由预设特征提取模型根据输入的样本图像确定；根据图像特征和质量分数，计算待搜索图像与各库图像之间的相似度；根据相似度，确定与待搜索图像关联的目标库图像。该方法的特征提取模型在训练时结合图像质量进行自适应间隔调整，得到的特征提取模型对输入的待搜索图像可以提取更准确的图像特征，进而在搜索图像时得到更准确的搜索结果。

技术领域

本公开涉及图像处理技术领域，尤其涉及一种图像搜索方法、装置、电子设备及计算机可读存储介质。

背景技术

图像搜索，是通过搜索图像文本或者视觉特征，为用户提供互联网上相关图形图像资料检索服务的专业搜索引擎系统，是搜索引擎的一种细分。其中一种图像搜索方法为通过上传图片搜索与该图片相同或相似的图片。

然而，相关技术中以图搜图的方法仅对图像特征进行描述，没有考虑到图片的质量不一致的问题，对质量不一致的图片没有针对性的优化，得到的图像特征分布不够准确，容易导致图像搜索的结果不够准确。

发明内容

有鉴于此，本公开实施例提供了一种图像搜索方法、装置、电子设备及计算机可读存储介质，以解决现有技术中图像搜索的结果不够准确的问题。

本公开实施例的第一方面，提供了一种图像搜索方法，包括：

获取待搜索图像；

将所述待搜索图像输入特征提取模型，得到所述待搜索图像的图像特征和质量分数；其中，所述特征提取模型是基于样本图像质量分数进行自适应间隔训练确定，所述样本图像质量分数由预设特征提取模型根据输入的样本图像确定；

根据所述图像特征和所述质量分数，计算所述待搜索图像与各库图像之间的相似度；

根据所述相似度，确定与所述待搜索图像关联的目标库图像。

本公开实施例的第二方面，提供了一种图像搜索装置，包括：

获取模块，用于获取待搜索图像；

特征提取模块，用于将所述待搜索图像输入特征提取模型，得到所述待搜索图像的图像特征和质量分数；其中，所述特征提取模型是基于样本图像质量分数进行自适应间隔训练确定，所述样本图像质量分数由预设特征提取模型根据输入的样本图像确定；

相似度计算模块，用于根据所述图像特征和所述质量分数，计算所述待搜索图像与各库图像之间的相似度；

搜索模块，用于根据所述相似度，确定与所述待搜索图像关联的目标库图像。

本公开实施例的第三方面，提供了一种电子设备，包括存储器、处理器以及存储在存储器中并且可在处理器上运行的计算机程序，该处理器执行计算机程序时实现上述方法的步骤。

本公开实施例的第四方面，提供了一种计算机可读存储介质，该计算机可读存储介质存储有计算机程序，该计算机程序被处理器执行时实现上述方法的步骤。

本公开实施例与现有技术相比存在的有益效果是：通过将获取的待搜索图像输入经训练确定的特征提取模型，得到待搜索图像的图像特征和质量分数；其中，特征提取模型是基于样本图像质量分数进行自适应间隔进行训练得到的，样本图像质量分数由预设特征提取模型根据样本图像确定；然后根据图像特征、质量分数在库图像中基于相似度的方式搜索得到与待搜索图像关联的目标库图像。上述方法的特征提取模型在训练时考虑了样本图像的图像质量，对样本图像确定对应的质量分数，并以图像质量分数为基础对预设特征提取模型提取的图像特征进行自适应间隔调整，从而训练得到特征提取模型，因此特征提取模型可以对输入的待搜索图像提取更准确的图像特征，进而在搜索图像时得到更准确的搜索结果。