一种残膜回收机防缠绕挑膜装置的制 一种秧草收获机用电力驱动行走机构

一种模型训练方法及系统、人脸图像去光方法及系统与流程

2022-09-15 06:47:11 来源:中国专利 TAG:


1.本发明涉及深度学习模型技术领域,具体涉及一种模型训练方法及系统、人脸图像去光方法及系统、电子设备及可读存储介质。


背景技术:

2.随着硬件和网络的发展,拍照变得越来越便捷,然而受限于拍摄环境,拍出来的照片往往很容易受到光线和阴影的不良影响,例如,光线过强造成的过曝光,或者不恰当的阴影位置等等。修图师可以通过一系列的专业工具来解决这个问题,但是这样做的成本较高。
3.当前针对人脸图像去光方法(包括去高光和/或去阴影)已经有一些进展,但是还存在以下几个问题:1)人脸图像去光方法依赖人脸图像去光模型的训练,而人脸图像去光模型的训练没有可获得的大型公开数据集,效果比较好的方法数据往往是基于昂贵的light stage(灯光舞台)采集的,导致算法的普及性不强;2)现有技术通过从2d人脸图像估计出人脸3d形状,然后对估计出的人脸3d形状进行去光,最后再渲染回2d图像,得到光照均匀的人脸图像,由于2d到3d的转换过程中会存在很大误差,所以影响最终的人脸图像去光效果。


技术实现要素:

4.为至少在一定程度上克服相关技术中存在的问题,本发明提供一种模型训练方法及系统、人脸图像去光方法及系统、电子设备及可读存储介质,以解决现有技术中人脸图像去光方法依赖人脸图像去光模型的训练,而人脸图像去光模型的训练需要使用昂贵的设备进行数据采集,导致成本高、算法普及性不强的问题。
5.根据本发明实施例的第一方面,提供一种模型训练方法,包括:
6.获取高清人脸图像数据集;
7.从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并将筛选出的光照均匀的人脸图像存储在真实人脸图像集中;
8.对筛选出的光照均匀的人脸图像重光照,得到带有阴影和高光的人脸图像,生成人脸重光照图像集;
9.将所述人脸重光照图像集中的人脸图像及真实人脸图像集中的人脸图像输入到预先构建的网络模型中进行训练,直至模型收敛,得到人脸图像去光模型,所述人脸图像去光模型用于对输入的人脸图像进行去光处理,得到光照均匀的人脸图像。
10.进一步地,所述从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,包括:
11.对所述高清人脸图像数据集中的每张高清人脸图像,进行人脸分割,获得人脸图像中的人脸区域;
12.对每个分割出的人脸区域,根据所述人脸区域中的高光像素和阴影像素占比,判断该人脸区域对应的人脸图像是否为光照均匀的人脸图像。
13.进一步地,所述根据所述人脸区域中的高光像素和阴影像素占比,判断该人脸区域对应的人脸图像是否为光照均匀的人脸图像,包括:
14.在所述人脸区域中,将像素值高于第一预设像素值的像素标记为高光像素,将像素值低于第二预设像素值的像素标记为阴影像素;所述第二预设像素值低于第一预设像素值;
15.计算阴影像素的数量和高光像素的数量两者之和占所述人脸区域所有像素数量的比重;
16.若所述比重低于阈值,则将该人脸区域对应的人脸图像标记为光照均匀的人脸图像。
17.进一步地,所述对筛选出的光照均匀的人脸图像重光照,包括:
18.随机挑选球谐函数系数,将所述球谐函数系数所对应的光照方向和光照强度添加到筛选出的光照均匀的人脸图像中,通过调节所述球谐函数系数的缩放尺度,得到带有阴影和高光的人脸图像;
19.将剪影数据作为外部阴影的轮廓或者通过添加柏林噪声仿造无固定形状的外部物体造成的阴影,添加到所述带有阴影和高光的人脸图像中。
20.进一步地,所述将所述人脸重光照图像集中的人脸图像及真实人脸图像集中的人脸图像输入到预先构建的网络模型中进行训练,包括:
21.定义所述网络模型的损失函数;
22.更新所述网络模型的生成器和判别器,直至模型收敛;
23.其中,所述生成器用于接收重光照后的人脸图像,并对重光照后的人脸图像进行去光处理,生成光照均匀的人脸图像;
24.所述判别器用于接收生成器生成的人脸图像,或者真实的光照均匀的人脸图像,输出所述人脸图像属于真实的光照均匀的人脸图像的置信度。
25.进一步地,所述网络模型的损失函数,至少包括:
26.用于优化生成器的基于标准差的正则项;
27.所述正则项用于约束生成器生成的人脸图像的光照效果。
28.进一步地,所述更新所述网络模型的生成器和判别器,包括:
29.更新生成器,包括:
30.固定判别器的网络参数,将所述人脸重光照图像集中的人脸图像分批送入生成器,输出光照均匀的人脸图像;
31.根据判别器输出的该光照均匀的人脸图像的置信度,计算损失函数值,并对生成器各层的模型参数进行求导,得到生成器各层的梯度;
32.根据所述梯度,使用梯度下降算法和反向传播算法更新生成器的网络参数;
33.更新判别器,包括:
34.将生成器生成的人脸图像和真实人脸图像集中的人脸图像,轮流输入到判别器中,分别得到输入的人脸图像属于真实的光照均匀的人脸图像的第一置信度和第二置信度;
35.根据所述第一置信度和第二置信度,计算损失函数值,并使用链式法则计算判别器各个层的梯度;
36.根据所述梯度,使用梯度下降算法和反向传播算法更新判别器的网络参数。
37.进一步地,所述方法,还包括:
38.设置所述生成器和判别器的优化迭代次数,所述优化迭代次数达到预设次数,损失函数值不再减小,且测试集上测得的结构相似度不再提升时,判定模型收敛,将此时的生成器确定为人脸图像去光模型;
39.所述测试集中存储有多张人脸图像及每张人脸图像对应的真值;所述真值是指人脸图像未被重光照时的像素值;
40.所述结构相似度是指测试集中的人脸图像输入到生成器后,生成光照均匀的人脸图像,该光照均匀的人脸图像的预测值与对应真值的相似度;
41.所述预测值是指该光照均匀的人脸图像的像素值。
42.根据本发明实施例的第二方面,提供一种人脸图像去光方法,包括:
43.将实时采集的人脸图像进行人脸对齐,以识别出人脸区域,并将识别出的人脸区域对齐到指定位置;
44.将人脸对齐后的人脸图像,输入到所述人脸图像去光模型中进行去光处理,得到光照均匀的人脸图像;所述人脸图像去光模型为利用权利要求1~8中任一项所述的模型训练方法得到的。
45.根据本发明实施例的第三方面,提供一种模型训练系统,包括:
46.获取模块,用于获取高清人脸图像数据集;
47.筛选模块,用于从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并将筛选出的光照均匀的人脸图像存储在真实人脸图像集中;
48.重光照模块,用于对筛选出的光照均匀的人脸图像重光照,生成人脸重光照图像集;
49.训练模块,用于将所述人脸重光照图像集中的人脸图像及真实人脸图像集中的人脸图像输入到预先构建的网络模型中进行训练,直至模型收敛,得到人脸图像去光模型,所述人脸图像去光模型,用于对输入的人脸图像进行去光处理,得到光照均匀的人脸图像。
50.进一步地,所述网络模型,包括:
51.生成器,用于接收重光照后的人脸图像,并对重光照后的人脸图像进行去光处理,生成光照均匀的人脸图像;
52.判别器,用于接收生成器生成的人脸图像,或者真实的光照均匀的人脸图像,输出所述人脸图像属于真实的光照均匀的人脸图像的置信度。
53.进一步地,所述生成器,包括:
54.编码器,用于对重光照后的人脸图像进行特征提取,得到去除光照影响的人脸特征;
55.解码器,用于将去除光照影响的人脸特征,还原为光照均匀的人脸图像。
56.进一步地,所述判别器,包括:
57.卷积层,用于对输入的光照均匀的人脸图像进行一次特征提取;
58.残差模块,用于对卷积层提取的特征进行二次特征提取和下采样;
59.全连接层,用于根据残差模块输出的特征,计算输入的光照均匀的人脸图像属于真实的光照均匀的人脸图像的置信度,并输出所述置信度。
60.根据本发明实施例的第四方面,提供一种人脸图像去光系统,包括:
61.对齐模块,用于将实时采集的人脸图像进行人脸对齐,以识别出人脸区域,并将识别出的人脸区域对齐到指定位置;
62.去光模块,用于将人脸对齐后的人脸图像,输入到所述人脸图像去光模型中进行去光处理,得到光照均匀的人脸图像;所述人脸图像去光模型为利用权利要求1~8中任一项所述的模型训练方法得到的。
63.根据本发明实施例的第五方面,提供一种电子设备,包括:处理器和存储器,其中,
64.所述存储器中存储有可供处理器调用的计算机程序;
65.所述处理器通过调用所述存储器中存储的计算机程序,执行上述的人脸图像去光模型的训练方法,和/或,人脸图像去光方法。
66.根据本发明实施例的第六方面,提供一种计算机可读存储介质,其上存储有可擦写的计算机程序;
67.当所述计算机程序在计算机设备上运行时,使得所述计算机设备执行上述的人脸图像去光模型的训练方法,和/或,人脸图像去光方法。
68.本发明的实施例提供的技术方案可以包括以下有益效果:
69.通过获取高清人脸图像数据集,从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并对筛选出的光照均匀的人脸图像重光照,生成人脸重光照图像集,实现了模型训练数据的自生成,摆脱了对昂贵的数据采集设备的依赖,成本低、算法普及性更强。
70.进一步地,本发明提供的技术方案,无需2d到3d的转换,仅通过2d人脸图像就可以实现对人脸图像的去光,大大提升了人脸图像的去光效果。
71.应当理解的是,以上的一般描述和后文的细节描述仅是示例性和解释性的,并不能限制本发明。
附图说明
72.此处的附图被并入说明书中并构成本说明书的一部分,示出了符合本发明的实施例,并与说明书一起用于解释本发明的原理。
73.图1是根据一示例性实施例示出的一种模型训练方法的流程图;
74.图2是根据一示例性实施例示出的生成器的结构示意图;
75.图3是根据一示例性实施例示出的gan块的结构示意图;
76.图4是根据一示例性实施例示出的判别器的结构示意图;
77.图5是根据一示例性实施例示出的残差模块的结构示意图;
78.图6是根据一示例性实施例示出的一种模型训练系统的示意框图;
79.图7是根据一示例性实施例示出的一种人脸图像去光系统的示意框图。
具体实施方式
80.这里将详细地对示例性实施例进行说明,其示例表示在附图中。下面的描述涉及附图时,除非另有表示,不同附图中的相同数字表示相同或相似的要素。以下示例性实施例中所描述的实施方式并不代表与本发明相一致的所有实施方式。相反,它们仅是与如所附权利要求书中所详述的、本发明的一些方面相一致的装置和方法的例子。
81.如前面背景技术,现有技术中人脸图像去光方法依赖人脸图像去光模型的训练,而人脸图像去光模型的训练需要使用昂贵的设备进行数据采集,导致成本高、算法普及性不强。为了解决这一技术问题,本发明提供了以下几个示例性实施例。
82.需要说明的是,本发明各实施例所提及的“去光”是指去除当前人脸图像中的高光(过曝光),和/或,阴影;“重光照”是指将高光(过曝光),和/或,阴影添加到当前人脸图像中;“真实的光照均匀的人脸图像”是指真实人脸图像集中的人脸图像。
83.图1是根据一示例性实施例示出的一种人脸图像去光模型的训练方法的流程图,如图1所示,该方法包括:
84.步骤s11、获取高清人脸图像数据集;
85.步骤s12、从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并将筛选出的光照均匀的人脸图像存储在真实人脸图像集中;
86.步骤s13、对筛选出的光照均匀的人脸图像重光照,得到带有阴影和高光的人脸图像,生成人脸重光照图像集;
87.步骤s14、将所述人脸重光照图像集中的人脸图像及真实人脸图像集中的人脸图像输入到预先构建的网络模型中进行训练,直至模型收敛,得到人脸图像去光模型,所述人脸图像去光模型,用于对输入的人脸图像进行去光处理,得到光照均匀的人脸图像。
88.需要说明的是,本实施例提供的技术方案,适用于电脑端,和/或,移动终端,适用的场景包括但不限于:当前人脸图像中的高光(过曝光),和/或,阴影。
89.步骤s11中“获取高清人脸图像数据集”,具体为:
90.从互联网上获取公开的ffhq和celeba-hq人脸图像数据集。
91.其中,ffhq全称flickr-faces-hight-quality(flickr-faces-hq)是作为生成对抗网络(gan)的基准创建的,也用于style gan的训练数据集中,于2019年开源。ffhq是一个高质量的人脸数据集,包含1024x1024分辨率的70000张png格式高清人脸图像。
92.所述高清即为分辨率大于1024x1024的人脸图像。
93.celeba是celebfaces attribute的缩写,即名人人脸属性数据集,其包含10,177个名人身份的202,599张人脸图片,每张图片都做好了特征标记,包含人脸bbox标注框、5个人脸特征点坐标以及40个属性标记,celeba由香港中文大学开放提供。
94.步骤s12中“从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像”,包括:
95.对所述高清人脸图像数据集中的每张高清人脸图像,进行人脸分割,获得人脸图像中的人脸区域;
96.在所述人脸区域中,将像素值高于第一预设像素值的像素标记为高光像素,将像素值低于第二预设像素值的像素标记为阴影像素;所述第二预设像素值低于第一预设像素值;
97.计算阴影像素的数量和高光像素的数量两者之和占所述人脸区域所有像素数量的比重;
98.若所述比重低于阈值,则将该人脸区域对应的人脸图像标记为光照均匀的人脸图像。
99.在具体实践中,所述第一预设像素值、第二预设像素值、阈值根据用户需要进行设置,或者,根据实验数据进行设置,或者,根据历史经验值进行设置。例如,将第一预设像素
值设定为200,第二预设像素值设定为114。
100.步骤s13中“对筛选出的光照均匀的人脸图像重光照,生成人脸重光照图像集”,包括:
101.随机挑选球谐函数系数,将所述球谐函数系数所对应的光照方向和光照强度添加到筛选出的光照均匀的人脸图像中,通过调节所述球谐函数系数的缩放尺度,得到带有阴影和高光的人脸图像;
102.将剪影数据作为外部阴影的轮廓或者通过添加柏林噪声仿造无固定形状的外部物体造成的阴影,添加到所述带有阴影和高光的人脸图像中。
103.需要说明的是,球谐光照在现代图形渲染领域有着广泛的应用,可快速模拟复杂的实时光照。球谐光照是一种光照算法,这种算法的内核是定义在球面上的特殊函数:球谐函数,它可以对光照进行捕捉并在之后进行重新光照并且可以实时展示全局光照风格的区域光源与软阴影。
104.球谐函数有9个系数,这9个系数的大小共同决定当前的光照方向和光照强度。这9个系数同时乘以缩放尺度可以调节光照强度(光照方向不变,如果每个系数单独调整,光照方向才会改变),该缩放尺度大于1,代表光的强度增加,小于1代表光的强度减小。调节所述球谐函数系数的缩放尺度,就可以得到带有阴影和高光的人脸图像。
105.由于本技术选用的人脸图像为彩色图像,rgb三通道每个通道对应9个系数,那么就组成3*9的系数矩阵。
106.将这3*9系数矩阵和筛选出的光照均匀的人脸图像输入到学习模型中,去学习这个3*9系数矩阵对应的光照方向和光照强度,得到带有阴影和高光的人脸图像。
107.可以理解的是,对筛选出的光照均匀的人脸图像重光照,不仅要模拟自然光对人脸图像造成的阴影(通过上述调节球谐函数系数实现),还要模拟环境中外部物体对人脸图像造成的阴影。
108.模拟环境中外部物体对人脸图像造成的阴影,有两种方法:
109.1、在网上公开的剪影数据集中,随机挑选一个或多个剪影,仿造固定形状的外部物体造成的阴影,添加到所述带有阴影和高光的人脸图像中(通过调用融合函数,将剪影数据和人脸图像数据进行融合)。
110.2、通过噪声函数生成柏林噪声,仿造无固定形状的外部物体造成的阴影,添加到所述带有阴影和高光的人脸图像中(通过调用融合函数,将剪影数据和人脸图像数据进行融合)。
111.步骤s14中“将所述人脸重光照图像集中的人脸图像及真实人脸图像集中的人脸图像输入到预先构建的网络模型中进行训练”,包括:
112.定义所述网络模型的损失函数;
113.更新所述网络模型的生成器和判别器,直至模型收敛;
114.具体实践中,基于深度学习框架(包括但不限于:pytorch深度学习框架、tensorflow深度学习框架)构建所述网络模型,所述网络模型包括:
115.生成器,用于接收重光照后的人脸图像,并对重光照后的人脸图像进行去光处理,生成光照均匀的人脸图像;
116.判别器,用于接收生成器生成的人脸图像,或者真实的光照均匀的人脸图像,输出
所述人脸图像属于真实的光照均匀的人脸图像的置信度。
117.参见图2,所述生成器,包括:
118.编码器,用于对重光照后的人脸图像进行特征提取,得到去除光照影响的人脸特征;
119.解码器,用于将去除光照影响的人脸特征,还原为光照均匀的人脸图像。
120.参见图2,编码器包括多个卷积层conv和一个全连接层fc,其中每个卷积层conv的参数可以设定为如下表1所示(表1中步长是指每次卷积核移动的步幅,输出尺寸是指卷积后输出的向量维度):
121.名字卷积核大小输入通道数输出通道数步长输出尺寸conv0131611024*1024conv1316322512*512conv2332642256*256conv33641282128*128conv43128256264*64conv53256512232*32conv63512512216*16conv7351251228*8conv8351251224*4linear-512*4*4512-512
122.表1
123.可以理解的是,编码器中的卷积层conv主要用于特征提取,全连接层fc主要用于特征降维,将卷积层conv提取的特征降维到长度为512。
124.参见图2,编码器最终的输出为长度为512的特征向量z,每个卷积层conv的输出也会被保存下来,通过跨层连接提供给解码器。
125.编码器输出的长度为512的特征向量z通过一个由8个全连接层组成的映射网络mapping network生成长度为512的特征向量w。
126.编码器包括const矩阵和多个gan块,其中,固定矩阵是const矩阵,为一个4
×4×
512大小的固定矩阵,所有gan块结构相同,如下图3所示。
127.图3中,a代表长度为512的特征向量w,b代表编码器每个卷积层conv输出的中间结果。
128.进一步地,参见图4,所述判别器,包括:
129.一个卷积层conv,用于对输入的光照均匀的人脸图像进行一次特征提取;
130.多个残差模块residual block,用于对卷积层提取的特征进行二次特征提取和下采样;
131.全连接层(也即线性层linear),用于根据残差模块输出的特征,计算输入的光照均匀的人脸图像属于真实的光照均匀的人脸图像的置信度,并输出所述置信度。
132.其中,所有残差模块residual block的结构相同,如图5所示,每个残差模块由两个卷积和一个旁路卷积构成,最后对两路卷积输入求和。图5中,激活函数的类型可以为leakyrelu。
133.进一步地,所述定义所述网络模型的损失函数,至少包括:
134.1)重建损失函数:该损失函数仅参与优化生成器;
135.其中,λ
l1
是手动指定的超参数,是指人脸重光照图像集中的人脸图像,y是指真实的光照均匀的人脸图像,||
·
||1是指l1范数。
136.2)生成对抗损失函数,包括用于优化生成器的损失函数:
[0137][0138]
用于优化判别器的损失函数:
[0139]
其中,d(y)是指输入判别器的图像y是真实的光照均匀的人脸图像的置信度;是指输入判别器的图像是真实的光照均匀的人脸图像的置信度,其值越高,证明是真实的光照均匀的人脸图像的可能性越大。
[0140]
3)感知损失函数:该损失函数仅参与优化生成器;
[0141]
其中,φ是指提取的感知特征,在本实施例明中,使用vgg-19网络的卷积层conv1到卷积层conv5层进行特征提取。
[0142]
4)基于标准差的正则项:该损失函数仅参与优化生成器;
[0143]
其中,λ
reg
是指定的超参数,std(
·
)是指计算标准差,relu(x)=max(0,x)。
[0144]
综上,总的损失函数为:l=l
recon
l
gan
l
per
l
reg

[0145]
其中,正则项用于约束生成器生成的人脸图像的光照效果。
[0146]
可以理解的是,一个质量较高的光照均匀的人脸图像,像素值位于0到255之间,其均值不一,但是标准差均在20到30之间。因此本实施例提出基于标准差的约束项,对生成器输出的人脸图像的标准差进行约束,最终提升去光的效果,并对高光有明显的压制。
[0147]
步骤s14中“更新所述网络模型的生成器和判别器”,包括:
[0148]
更新生成器,包括:
[0149]
固定判别器的网络参数,将所述人脸重光照图像集中的人脸图像分批送入生成器,输出光照均匀的人脸图像;
[0150]
根据判别器输出的该光照均匀的人脸图像的置信度,计算损失函数值,并使用链式法则计算生成器各个层的梯度(链式法则即对生成器各层的模型参数进行求导,得到生成器各层的梯度);
[0151]
根据所述梯度,使用梯度下降算法和反向传播算法更新生成器的网络参数;
[0152]
更新判别器,包括:
[0153]
将生成器生成的人脸图像和真实人脸图像集中的人脸图像,轮流输入到判别器中,分别得到输入的人脸图像属于真实的光照均匀的人脸图像的第一置信度和第二置信度;
[0154]
根据所述第一置信度和第二置信度,计算损失函数值,并使用链式法则计算判别器各个层的梯度;
[0155]
根据所述梯度,使用梯度下降算法和反向传播算法更新判别器的网络参数。
[0156]
进一步地,设置所述生成器和判别器的优化迭代次数,所述优化迭代次数达到预设次数,损失函数值不再减小,且测试集上测得的结构相似度不再提升时,判定模型收敛,将此时的生成器确定为人脸图像去光模型。
[0157]
所述测试集中存储有多张人脸图像及每张人脸图像对应的真值;所述真值是指人脸图像未被重光照时的像素值;
[0158]
所述结构相似度是指测试集中的人脸图像输入到生成器后,生成光照均匀的人脸图像,该光照均匀的人脸图像的预测值与对应真值的相似度;
[0159]
所述预测值是指该光照均匀的人脸图像的像素值。
[0160]
在具体实践中,优化迭代次数根据用户需要进行设置,或者,根据实验数据进行设置,或者,根据历史经验值进行设置,例如,设置为100000次。
[0161]
可以理解的是,本实施例提供的技术方案,通过获取高清人脸图像数据集,从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并对筛选出的光照均匀的人脸图像重光照,生成人脸重光照图像集,实现了模型训练数据的自生成,摆脱了对昂贵的数据采集设备的依赖,成本低、算法普及性更强。
[0162]
进一步地,本实施例提供的技术方案,无需2d到3d的转换,仅通过2d人脸图像就可以实现对人脸图像的去光,大大提升了人脸图像的去光效果。
[0163]
在具体实践中,通过上述的模型训练方法训练出的人脸图像去光模型可以用于人脸图像去光。根据一示例性实施例示出的一种人脸图像去光方法,包括:
[0164]
将实时采集的人脸图像进行人脸对齐,以识别出人脸区域,并将识别出的人脸区域对齐到指定位置;
[0165]
将人脸对齐后的人脸图像,输入到所述人脸图像去光模型中进行去光处理,得到光照均匀的人脸图像;所述人脸图像去光模型为利用上述的模型训练方法得到的。
[0166]
可以理解的是,本实施例提供的技术方案,由于需要通过实施例一所述的人脸图像去光模型的训练方法,得到人脸图像去光模型,而实施例一所述的人脸图像去光模型的训练方法通过获取高清人脸图像数据集,从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并对筛选出的光照均匀的人脸图像重光照,生成人脸重光照图像集,实现了模型训练数据的自生成,摆脱了对昂贵的数据采集设备的依赖,成本低、算法普及性更强。
[0167]
进一步地,本实施例提供的技术方案,无需2d到3d的转换,仅通过2d人脸图像就可以实现对人脸图像的去光,大大提升了人脸图像的去光效果。
[0168]
图6是根据一示例性实施例示出的一种模型训练系统100的示意框图,如图6所示,该系统100包括:
[0169]
获取模块101,用于获取高清人脸图像数据集;
[0170]
筛选模块102,用于从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并将筛选出的光照均匀的人脸图像存储在真实人脸图像集中;
[0171]
重光照模块103,用于对筛选出的光照均匀的人脸图像重光照,得到带有阴影和高光的人脸图像,生成人脸重光照图像集;
[0172]
训练模块104,用于将所述人脸重光照图像集中的人脸图像及真实人脸图像集中的人脸图像输入到预先构建的网络模型中进行训练,直至模型收敛,得到人脸图像去光模
型,所述人脸图像去光模型,用于对输入的人脸图像进行去光处理,得到光照均匀的人脸图像。
[0173]
需要说明的是,由于上述各模块的实现方式及有益效果可参考前述实施例中对应步骤的详细阐述,本实施例不再赘述。
[0174]
可以理解的是,本实施例提供的技术方案,通过获取高清人脸图像数据集,从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并对筛选出的光照均匀的人脸图像重光照,生成人脸重光照图像集,实现了模型训练数据的自生成,摆脱了对昂贵的数据采集设备的依赖,成本低、算法普及性更强。
[0175]
进一步地,本实施例提供的技术方案,无需2d到3d的转换,仅通过2d人脸图像就可以实现对人脸图像的去光,大大提升了人脸图像的去光效果。
[0176]
图7是根据一示例性实施例示出的一种人脸图像去光系统200的示意框图,如图7所示,该系统200包括:
[0177]
对齐模块201,用于将实时采集的人脸图像进行人脸对齐,以识别出人脸区域,并将识别出的人脸区域对齐到指定位置;
[0178]
去光模块202,用于将人脸对齐后的人脸图像,输入到所述人脸图像去光模型中进行去光处理,得到光照均匀的人脸图像;所述人脸图像去光模型为利用上述的模型训练方法得到的。
[0179]
可以理解的是,本实施例提供的技术方案,通过获取高清人脸图像数据集,从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并对筛选出的光照均匀的人脸图像重光照,生成人脸重光照图像集,实现了模型训练数据的自生成,摆脱了对昂贵的数据采集设备的依赖,成本低、算法普及性更强。
[0180]
进一步地,本实施例提供的技术方案,无需2d到3d的转换,仅通过2d人脸图像就可以实现对人脸图像的去光,大大提升了人脸图像的去光效果。
[0181]
根据一示例性实施例示出的一种电子设备,包括:处理器和存储器,其中,所述存储器中存储有可供处理器调用的计算机程序;
[0182]
所述处理器通过调用所述存储器中存储的计算机程序,执行上述的人脸图像去光模型的训练方法,和/或,人脸图像去光方法。
[0183]
需要说明的是,电子设备包括但不限于:智能终端(例如,手机、平板电脑、智能手表等)和计算机设备。
[0184]
所述处理器包括但不限于:cpu、单片机、plc控制器、fpga控制器等。
[0185]
所述存储器可以包括易失性存储器形式的计算机系统可读介质,例如随机存取存储器(ram)和/或高速缓存存储器;还可以包括其它可移动/不可移动的、易失性/非易失性计算机系统存储介质。存储器可以包括至少一个程序产品,该程序产品具有一组(例如至少一个)程序模块,这些程序模块被配置以执行本发明各实施例的功能。
[0186]
可以理解的是,本实施例提供的技术方案,通过获取高清人脸图像数据集,从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并对筛选出的光照均匀的人脸图像重光照,生成人脸重光照图像集,实现了模型训练数据的自生成,摆脱了对昂贵的数据采集设备的依赖,成本低、算法普及性更强。
[0187]
进一步地,本实施例提供的技术方案,无需2d到3d的转换,仅通过2d人脸图像就可
以实现对人脸图像的去光,大大提升了人脸图像的去光效果。
[0188]
根据一示例性实施例示出的一种计算机可读存储介质,其上存储有可擦写的计算机程序;
[0189]
当计算机程序在计算机设备上运行时,使得计算机设备执行上述的人脸图像去光模型的训练方法,和/或,人脸图像去光方法。
[0190]
本实施例公开的计算机可读存储介质包括但不限于:电、磁、光、电磁、红外线、或半导体的系统、装置或器件,或者任意以上的组合。计算机可读存储介质的更具体的例子(非穷举的列表)包括:具有一个或多个导线的电连接、便携式计算机磁盘、硬盘、随机存取存储器(ram)、只读存储器(rom)、可擦式可编程只读存储器(eprom或闪存)、光纤、便携式紧凑磁盘只读存储器(cd-rom)、光存储器件、磁存储器件、或者上述的任意合适的组合。在本发明中,计算机可读存储介质可以是任何包含或存储程序的有形介质,该程序可以被指令执行系统、装置或者器件使用或者与其结合使用。
[0191]
可以理解的是,本实施例提供的技术方案,通过获取高清人脸图像数据集,从所述高清人脸图像数据集中,筛选出光照均匀的人脸图像,并对筛选出的光照均匀的人脸图像重光照,生成人脸重光照图像集,实现了模型训练数据的自生成,摆脱了对昂贵的数据采集设备的依赖,成本低、算法普及性更强。
[0192]
进一步地,本实施例提供的技术方案,无需2d到3d的转换,仅通过2d人脸图像就可以实现对人脸图像的去光,大大提升了人脸图像的去光效果。
[0193]
可以理解的是,上述各实施例中相同或相似部分可以相互参考,在一些实施例中未详细说明的内容可以参见其他实施例中相同或相似的内容。
[0194]
需要说明的是,在本发明的描述中,术语“第一”、“第二”等仅用于描述目的,而不能理解为指示或暗示相对重要性。此外,在本发明的描述中,除非另有说明,“多个”的含义是指至少两个。
[0195]
流程图中或在此以其他方式描述的任何过程或方法描述可以被理解为,表示包括一个或更多个用于实现特定逻辑功能或过程的步骤的可执行指令的代码的模块、片段或部分,并且本发明的优选实施方式的范围包括另外的实现,其中可以不按所示出或讨论的顺序,包括根据所涉及的功能按基本同时的方式或按相反的顺序,来执行功能,这应被本发明的实施例所属技术领域的技术人员所理解。
[0196]
应当理解,本发明的各部分可以用硬件、软件、固件或它们的组合来实现。在上述实施方式中,多个步骤或方法可以用存储在存储器中且由合适的指令执行系统执行的软件或固件来实现。例如,如果用硬件来实现,和在另一实施方式中一样,可用本领域公知的下列技术中的任一项或他们的组合来实现:具有用于对数据信号实现逻辑功能的逻辑门电路的离散逻辑电路,具有合适的组合逻辑门电路的专用集成电路,可编程门阵列(pga),现场可编程门阵列(fpga)等。
[0197]
本技术领域的普通技术人员可以理解实现上述实施例方法携带的全部或部分步骤是可以通过程序来指令相关的硬件完成,所述的程序可以存储于一种计算机可读存储介质中,该程序在执行时,包括方法实施例的步骤之一或其组合。
[0198]
此外,在本发明各个实施例中的各功能单元可以集成在一个处理模块中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个模块中。上述集成的模
块既可以采用硬件的形式实现,也可以采用软件功能模块的形式实现。所述集成的模块如果以软件功能模块的形式实现并作为独立的产品销售或使用时,也可以存储在一个计算机可读取存储介质中。
[0199]
上述提到的存储介质可以是只读存储器,磁盘或光盘等。
[0200]
在本说明书的描述中,参考术语“一个实施例”、“一些实施例”、“示例”、“具体示例”、或“一些示例”等的描述意指结合该实施例或示例描述的具体特征、结构、材料或者特点包含于本发明的至少一个实施例或示例中。在本说明书中,对上述术语的示意性表述不一定指的是相同的实施例或示例。而且,描述的具体特征、结构、材料或者特点可以在任何的一个或多个实施例或示例中以合适的方式结合。
[0201]
尽管上面已经示出和描述了本发明的实施例,可以理解的是,上述实施例是示例性的,不能理解为对本发明的限制,本领域的普通技术人员在本发明的范围内可以对上述实施例进行变化、修改、替换和变型。
再多了解一些

本文用于企业家、创业者技术爱好者查询,结果仅供参考。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表

相关文献