[发明专利]图片检索方法、电子设备及存储介质

说明书

本发明提供一种图片检索方法，所述方法包括：获取待检索人脸图片；获取至少一个类别中每个类别的图片；提取所述待检索人脸图片的特征；计算所述待检索人脸图片的特征与至少一个类别中每个类别的图片的特征的相似度；确定每个类别对应的相似度阈值；根据每个类别对应的相似度阈值及所述待检索人脸图片的特征与至少一个类别中每个类别的图片的特征的相似度，从每个类别的图片中，筛选每个类别中符合条件的图片；基于每个类别中符合条件的图片，输出检索结果。本发明还提供一种电子设备及存储介质。本发明能在海量数据中检索出与待检索人脸图片相似的多种类别的图片，提高了检索精度和准度。

技术领域

本发明涉及图片处理领域，尤其涉及一种图片检索方法、电子设备及存储介质。

背景技术

人脸识别技术在许多领域有着非常广泛的应用，如视频监控、智慧商业、门禁系统等。这样，数据库就会有上亿级的大规模人脸库，在如此庞大的数据库中人脸的姿态、人脸所处的环境因素也是有很大不同的，尤其体现在光照、分辨率、遮挡等因素。因此，在对人脸图片进行检索时，很难检索到复杂环境下的人脸图片(光线不好的人脸图片，戴帽子的人脸图片)，检索精度不高，这样就无法满足用户的检索需求。

发明内容

鉴于以上内容，有必要提供一种图片检索方法、电子设备及存储介质，能在海量数据中检索出与待检索人脸图片相似的多种类别的图片，提高了检索精度和准度。

一种图片检索方法，所述方法包括：

获取待检索人脸图片；

获取至少一个类别中每个类别的图片；

提取所述待检索人脸图片的特征；

计算所述待检索人脸图片的特征与至少一个类别中每个类别的图片的特征的相似度；

确定每个类别对应的相似度阈值；

根据每个类别对应的相似度阈值及所述待检索人脸图片的特征与至少一个类别中每个类别的图片的特征的相似度，从每个类别的图片中，筛选每个类别中符合条件的图片；

基于每个类别中符合条件的图片，输出检索结果。

根据本发明优选实施例，所述至少一个类别中不同类别对应不同姿态类型的图片，每个类别对应一种姿态类型。

根据本发明优选实施例，所述确定每个类别对应的相似度阈值包括：

利用训练好的分类模型，确定所述待检索人脸图片的类别；

将所述待检索人脸图片的类别对应的相似度阈值配置为第一阈值；

将所述至少一个类别中其他类别对应的相似度阈值配置为第二阈值，所述第一阈值大于第二阈值。

根据本发明优选实施例，所述确定每个类别对应的相似度阈值包括：

接收终端设备发送的在用户界面上对每个类别配置的相似度阈值，所述用户界面上包括每个类别对应的相似度阈值选项。

根据本发明优选实施例，所述根据每个类别对应的相似度阈值及所述待检索人脸图片的特征与至少一个类别中每个类别的图片的特征的相似度，从每个类别的图片中，筛选每个类别中符合条件的图片包括：

从每个类别的图片中，筛选出相似度大于每个类别对应的相似度阈值的图片作为每个类别中符合条件的图片。

根据本发明优选实施例，在提取所述待检索人脸图片的特征之前，所述方法还包括：

对所述待检索人脸图片进行人脸对齐及人脸的归一化。

根据本发明优选实施例，所述提取所述待检索人脸图片的特征包括：