【技术实现步骤摘要】
一种人脸质量增强模型的训练方法、图片搜索方法及装置
[0001]本专利技术涉及图片搜索
,尤其涉及一种人脸质量增强模型的训练方法、图片搜索方法及装置。
技术介绍
[0002]在对人脸图片进行搜索时,一般由用户上传包含人脸图像的搜索图片,并发起搜索请求(即:Query),图片搜索引擎基于该搜索图片在互联网上搜索相关的图片,同时,图片搜索引擎还可以识别出该搜索图片中的人物身份(例如:姓名)。
[0003]若搜索图片中的人脸图像尺寸太小(称作“小人脸”),则图片搜索引擎难以从搜索图片中提取到足够的特征,图片搜索引擎的性能将受到影响,无法获得准确的搜索结果。此时,则需要对搜索图片进行人脸质量增强,
[0004]但是,目前的人脸质量增强方法,难以拟合真实场景,在对小人脸图片进行搜索时,搜索效果依然不好,搜索结果的准确性不高。
技术实现思路
[0005]本专利技术实施例通过提供及一种人脸质量增强模型的训练方法、图片搜索方法及装置,解决了现有技术中在对小人脸图片进行图片搜索时,难以获得准确的搜索结果的 ...
【技术保护点】
【技术特征摘要】
1.一种人脸质量增强模型的训练方法,其特征在于,包括:构建训练数据,所述训练数据中包含多个人脸图片对,每个所述人脸图片对中均包含第一人脸图片和第二人脸图片,所述第一人脸图片和所述第二人脸图片中包含同一人物的人脸图像,所述第一人脸图片和第二人脸图片均为预设尺寸,所述第二人脸图片的分辨率等级高于所述第一人脸图片的分辨率等级,所述训练数据中的全部图片共对应有N+1个分辨率等级,N为正整数;构建初始训练模型,所述初始训练模型中包含N个模块,所述N个模块依次级联,每个模块均具有一输入端和一输出端,共有N个输入端和N个输出端,所述N个输入端各自对应的分辨率等级不同,所述N个输出端各自对应的分辨率等级不同;对于所述训练数据中的每个人脸图片对,从所述N个输入端中选出与所述第一人脸图片的分辨率等级对应的目标输入端,以及从所述N个输出端中选出与所述第二人脸图片的分辨率等级对应的目标输出端;将所述第一人脸图片作为所述目标输入端的输入,并将所述第二人脸图片作为所述目标输出端的输出,以对所述初始训练模型进行训练,获得人脸质量增强模型。2.如权利要求1所述的方法,其特征在于,所述构建训练数据,包括:获得M个人脸图片集合,每个人脸图片集合中均包含多张人脸图片,所述多张人脸图片的分辨率等级各不相同,同一人脸图片集合中的人脸图片均包含同一人物的人脸图像,所述多张人脸图片均为所述预设尺寸;针对每个所述人脸图片集合进行多次人脸图片提取,每次随机提出两张人脸图片,获得多个人脸图片对;并基于所述多个人脸图片对,构建所述训练数据。3.如权利要求2所述的方法,其特征在于,所述获得M个人脸图片集合,包括:获取P张原始图片,P为正整数;从所述P张原始图片中筛选出M张原始图片,所述M张原始图片中均包含小于所述预设尺寸的人脸图像;对所述M张原始图片分别进行图片搜索,获得所述M个图片集合;其中,所述M个图片集合与所述M张原始图片一一对应。4.一种图片搜索方法,其特征在于,包括:获取用户上传的第一搜索图片;确定所述第一搜索图片的分辨率等级;基于所述分辨率等级,从人脸质量增强模型中选择与所述分辨率等级对应的输入端作为目标输入端;其中,所述人脸质量增强模型由权利要求1~6任一所述的方法训练获得;将所述第一搜索图片从所述目标输入端输入到所述人脸质量增强模型中,以通过所述人脸质量增强模型对所述第一搜索图片进行人脸质量增强,获得第二搜索图片;对所述第二搜索图片进行图片搜索,获得搜索结果。5.一种人脸质量增强模型的训练装置,其特...
【专利技术属性】
技术研发人员:李天鹏,林明安,刘慧慧,周泽南,许静芳,
申请(专利权)人:北京搜狗科技发展有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。