利索能及
我要发布
收藏
专利号: 2021107301283
申请人: 北京百度网讯科技有限公司
专利类型:发明专利
专利状态:已下证
更新日期:2025-12-08
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种训练集生成方法,包括:

获取人脸图像数据集;

通过图像美学算法确定所述人脸图像数据集中每张人脸图像的美感得分;

通过人脸质量评价算法确定所述人脸图像数据集中每张人脸图像的人脸质量得分,所述人脸质量得分用于表示人脸特征的完整性;

通过图像质量评估算法确定所述人脸图像数据集中每张人脸图像的自适应质量得分,所述自适应得分用于弥补所述图像美学算法与所述人脸质量评价算法对人脸图像评价标准缺失的部分;

分别根据所述人脸图像数据集中每张人脸图像的美感得分、人脸质量得分和自适应质量得分,确定每张人脸图像的综合得分;

按照人脸图像的综合得分从大到小的顺序,选取预设数量的人脸图像,以生成用于训练人脸属性编辑模型的训练集;

所述分别根据所述人脸图像数据集中每张人脸图像的美感得分、人脸质量得分和自适应质量得分,确定每张人脸图像的综合得分,包括:针对每张人脸图像进行以下处理:

分别使用第一归一化参数、第二归一化参数、第三归一化参数对人脸图像的美感得分、人脸质量得分和自适应质量得分进行归一化;

将归一化后的人脸图像的美感得分、人脸质量得分和自适应质量得分通过预设权重参数进行加权求和,使用第四归一化参数对加权求和结果进行归一化,得到该人脸图像的综合得分;

所述预设权重参数通过以下方式获得:

通过预设数量组权重参数分别对所述人脸图像数据集中包括的人脸图像计算综合得分,以生成预设数量个测试训练集,其中,每组权重参数包括美感得分权重、人脸质量得分权重和自适应质量得分权重;

分别使用每个测试训练集对深度学习模型进行训练,得到每个测试训练集对应的测试模型;

通过每个测试模型对相同的人脸图像进行人脸属性编辑,根据每个测试模型输出的人脸编辑图像确定每个测试模型的测试结果;

通过每个测试模型的测试结果确定所述预设权重参数。

2.根据权利要求1所述的方法,其中,所述通过每个测试模型的测试结果确定所述预设权重参数,包括:将测试结果最好的测试模型对应的一组权重参数作为所述预设权重参数;或者,将测试模型分为第一类别和第二类别,所述第一类别的测试模型的测试结果优于所述第二类别的测试模型的测试结果;

将所述第一类别的测试模型对应的权重参数进行按位求平均值,得到第一权重向量;

将所述第二类别的测试模型对应的权重参数进行按位求平均值,得到第二权重向量;

根据所述第一权重向量和所述第二权重向量确定方向向量;

根据所述方向向量对所述第一权重向量进行优化,得到所述预设权重参数。

3.根据权利要求1所述的方法,所述方法还包括:每确定一张人脸图像的综合得分,根据该人脸图像的综合得分将该人脸图像加入人脸图像序列中,所述人脸图像序列包括的人脸图像按照综合得分大小排列。

4.一种人脸图像处理方法,包括:

获取待编辑的人脸图像;

将所述待编辑的人脸图像输入人脸属性编辑模型,所述人脸属性编辑模型基于训练集训练得到,所述训练集为基于权利要求1‑3中任一项所述的方法生成的训练集;

获取所述人脸属性编辑模型输出的进行人脸属性编辑后的人脸图像。

5.一种训练集生成装置,包括:

获取模块,用于获取人脸图像数据集;

第一确定模块,用于通过图像美学算法确定所述人脸图像数据集中每张人脸图像的美感得分;

第二确定模块,用于通过人脸质量评价算法确定所述人脸图像数据集中每张人脸图像的人脸质量得分,所述人脸质量得分用于表示人脸特征的完整性;

第三确定模块,用于通过图像质量评估算法确定所述人脸图像数据集中每张人脸图像的自适应质量得分,所述自适应得分用于弥补所述图像美学算法与所述人脸质量评价算法对人脸图像评价标准缺失的部分;

综合得分确定模块,用于分别根据所述人脸图像数据集中每张人脸图像的美感得分、人脸质量得分和自适应质量得分,确定每张人脸图像的综合得分;

生成模块,用于按照人脸图像的综合得分从大到小的顺序,选取预设数量的人脸图像,以生成用于训练人脸属性编辑模型的训练集;

所述综合得分确定模块,具体用于:

针对每张人脸图像进行以下处理:

分别使用第一归一化参数、第二归一化参数、第三归一化参数对人脸图像的美感得分、人脸质量得分和自适应质量得分进行归一化;

将归一化后的人脸图像的美感得分、人脸质量得分和自适应质量得分通过预设权重参数进行加权求和,使用第四归一化参数对加权求和结果进行归一化,得到该人脸图像的综合得分;

所述装置还包括:

所述生成模块,还用于通过预设数量组权重参数分别对所述人脸图像数据集中包括的人脸图像计算综合得分,以生成预设数量个测试训练集,其中,每组权重参数包括美感得分权重、人脸质量得分权重和自适应质量得分权重;

训练模块,用于分别使用每个测试训练集对深度学习模型进行训练,得到每个测试训练集对应的测试模型;

测试模块,用于通过每个测试模型对相同的人脸图像进行人脸属性编辑,根据每个测试模型输出的人脸编辑图像确定每个测试模型的测试结果;

第四确定模块,用于通过每个测试模型的测试结果确定所述预设权重参数。

6.根据权利要求5所述的装置,其中,所述第四确定模块,具体用于:将测试结果最好的测试模型对应的一组权重参数作为所述预设权重参数;或者,将测试模型分为第一类别和第二类别,所述第一类别的测试模型的测试结果优于所述第二类别的测试模型的测试结果;

将所述第一类别的测试模型对应的权重参数进行按位求平均值,得到第一权重向量;

将所述第二类别的测试模型对应的权重参数进行按位求平均值,得到第二权重向量;

根据所述第一权重向量和所述第二权重向量确定方向向量;

根据所述方向向量对所述第一权重向量进行优化,得到所述预设权重参数。

7.根据权利要求5所述的装置,所述装置还包括:加入模块,用于每确定一张人脸图像的综合得分,根据该人脸图像的综合得分将该人脸图像加入人脸图像序列中,所述人脸图像序列包括的人脸图像按照综合得分大小排列。

8.一种人脸图像处理装置,包括:

第一获取模块,用于获取待编辑的人脸图像;

输入模块,用于将所述待编辑的人脸图像输入人脸属性编辑模型,所述人脸属性编辑模型基于训练集训练得到,所述训练集为基于权利要求5‑7中任一项所述的装置生成的训练集;

第二获取模块,用于获取所述人脸属性编辑模型输出的进行人脸属性编辑后的人脸图像。

9.一种电子设备,包括:

至少一个处理器;以及

与所述至少一个处理器通信连接的存储器;其中,

所述存储器存储有可被所述至少一个处理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器能够执行权利要求1‑4中任一项所述的方法。

10.一种存储有计算机指令的非瞬时计算机可读存储介质,其中,所述计算机指令用于使所述计算机执行根据权利要求1‑4中任一项所述的方法。