[发明专利]一种图片筛选方法、装置、电子设备及存储介质

说明书

本申请实施例提供一种图片筛选方法、装置、电子设备及存储介质，方法包括：获取目标关键词，获取与目标关键词相匹配的目标图像识别模型，利用目标图像识别模型对获取的待筛选图片集中的图片进行识别，得到各图片的识别结果，识别结果用于指示图片是否与目标关键词相匹配，根据各图片的识别结果，输出待筛选图片集中与目标关键词相匹配的目标图片。通过本方案可以根据目标关键词从待筛选图片集中筛选出与目标关键词相匹配的图片，在使用时，可以根据要筛选出的图片中包含的特定内容设定目标关键词，进而可以根据目标关键词筛选出包含特定内容的图片，无需用户再一张一张的查找，节省了人力和时间。

技术领域

本申请涉及图像处理领域，尤其涉及一种图片筛选方法、装置、电子设备及存储介质。

背景技术

目前终端中一般都会设置有用于存储图片的图库，图库中的图片通常为用户通过相机拍摄的图片和用户下载的图片。

目前在对图库中的图片进行筛选时，只能通过拍摄日期、拍摄地点等拍照时自动生成的标签、或者图片的存储路径、下载路径等属性来对图库中的图片进行筛选。

而用户若想从图库中筛选出具有某一特定内容的图片(比如包含动物的图片、包含植物的图片、包含证件的图片、包含交通工具的图片等)，则只能手动的一张一张的对图库中的图片进行查找，会耗费过多的人力和时间。

发明内容

为了解决现有的手动对包含特定内容的图片进行筛选存在的耗费人力和时间过多的技术问题，本申请提供了一种图片筛选方法、装置、电子设备及存储介质。

第一方面，本申请提供了一种图片筛选方法，包括：

获取目标关键词；

获取与所述目标关键词相匹配的目标图像识别模型；

将待筛选图片集中的各图片分别输入所述目标图像识别模型，输出各图片对应的识别结果，所述识别结果用于指示图片是否与所述目标关键词相匹配；

根据所述各图片对应的识别结果对所述待筛选图片集进行筛选，输出所述待筛选图片集中与所述目标关键词相匹配的目标图片。

在一种可能的实现方式中，获取与所述目标关键词相匹配的目标图像筛选模型，包括：

从模型库中匹配出与所述目标关键词相匹配的目标图像识别模型，所述模型库包括多个预先训练好的图像识别模型，每个所述图像识别模型设置有对应的关键词标签，所述关键词标签用于与所述目标关键词进行匹配。

在一种可能的实现方式中，获取与所述目标关键词相匹配的目标图像筛选模型，还包括：

若未从所述模型库中匹配出与所述目标关键词相匹配的目标图像识别模型，则基于所述目标关键词确定对应的训练图片集和测试图片集；

利用所述训练图片集对初始模型进行训练，得到第一模型；

在所述第一模型对所述测试图片集的识别准确率不大于预设的准确率阈值的情况下，继续使用所述训练图片集对所述第一模型进行训练，以调整所述第一模型的参数，直至所述第一模型对所述测试图片集的识别准确率大于所述准确率阈值；

在所述第一模型对所述测试图片集的识别准确率大于所述准确率阈值的情况下，将所述第一模型作为与所述目标关键词相匹配的目标图像识别模型。

在一种可能的实现方式中，在所述第一模型对所述测试图片集的识别准确率大于所述准确率阈值的情况下，所述方法还包括：

基于所述目标关键词为所述第一模型设置关键词标签，并将设置有关键词标签的所述第一模型存储至模型库中。

第二方面，本申请实施例还提供了一种图片筛选装置，包括：

关键词获取模块，用于获取目标关键词；