[发明专利]一种图片搜索模型的选择方法和装置

说明书

本申请公开了一种图片搜索模型的选择方法和装置，该方法包括以下步骤：使用手工特征模型从参考图片库中检索与每张待检索图片相似度最高的预设数量的参考图片；获取手工特征模型的评测指标；在手工特征模型的评测指标满足预设条件的情况下，选择手工特征模型作为图片搜索模型；在手工特征模型的评测指标不满足预设条件的情况下，使用深度学习模型从参考图片库中检索与每张待检索图片相似度最高的预设数量的参考图片；获取深度学习模型的评测指标；在深度学习模型的评测指标满足预设条件的情况下，选择深度学习模型作为图片搜索模型。本申请实施例将图片匹配过程和指标评测过程进行集成，从而兼顾匹配效果和计算效率。

技术领域

本申请属于计算机技术领域，具体涉及一种图片搜索模型的选择方法和装置。

背景技术

随着互联网和数字经济的发展，数据正以指数爆炸的速度不断增加，图像作为一种重要的数据形式，广泛存在于社会的各行各业，例如社交、电商、医疗、交通和安防等。在这些行业，图片数量巨大，相似图片查询的任务，即以图搜图也越来越受到关注。

以图搜图是指根据图像内容搜索出相似内容的图像，数据规模的极速增长对以图搜图的计算效果和计算效率提出了更高的要求。而构建一个以图搜图系统需要解决两个问题：第一为提取图像特征，其次为构建成数据库并提供相似性搜索的功能。目前，以图搜图技术通常使用单一模型或解决方案，不能兼顾匹配效果和计算效率。

申请内容

本申请实施例的目的是提供一种图片搜索模型的选择方法和装置，以解决现有技术无法兼顾匹配效果和计算效率的缺陷。

为了解决上述技术问题，本申请是这样实现的：

第一方面，提供了一种图片搜索模型的选择方法，包括以下步骤：

设置参考图片库和待检索图片集，所述参考图片库包括多张参考图片，所述待检索图片集包括多张待检索图片，且所述参考图片库中存在与每张待检索图片匹配的目标图片；

使用手工特征模型从所述参考图片库中检索与每张待检索图片相似度最高的预设数量的参考图片；

根据所述预设数量的参考图片，获取所述手工特征模型的评测指标；

在所述手工特征模型的评测指标满足预设条件的情况下，选择所述手工特征模型作为图片搜索模型；

在所述手工特征模型的评测指标不满足预设条件的情况下，使用深度学习模型从所述参考图片库中检索与每张待检索图片相似度最高的预设数量的参考图片；

根据所述预设数量的参考图片，获取所述深度学习模型的评测指标；

在所述深度学习模型的评测指标满足预设条件的情况下，选择所述深度学习模型作为图片搜索模型。

第二方面，提供了一种图片搜索模型的选择装置，包括：

设置模块，用于设置参考图片库和待检索图片集，所述参考图片库包括多张参考图片，所述待检索图片集包括多张待检索图片，且所述参考图片库中存在与每张待检索图片匹配的目标图片；

第一检索模块，用于使用手工特征模型从所述参考图片库中检索与每张待检索图片相似度最高的预设数量的参考图片；

第一获取模块，用于根据所述预设数量的参考图片，获取所述手工特征模型的评测指标；

第一选择模块，用于在所述手工特征模型的评测指标满足预设条件的情况下，选择所述手工特征模型作为图片搜索模型；

第二检索模块，用于在所述手工特征模型的评测指标不满足预设条件的情况下，使用深度学习模型从所述参考图片库中检索与每张待检索图片相似度最高的预设数量的参考图片；