人像美白方法、装置、电子设备和可读存储介质.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN111784611A(43)申请公布日2020.10.16(21)申请号202010636778.7(22)申请日2020.07.03(71)申请人厦门美图之家科技有限公司地址361000福建省厦门市火炬高新区软件园华讯楼C区B1F-089(72)发明人周铭柯李启东邹嘉伟陈进山何恕预(74)专利代理机构北京超凡宏宇专利代理事务所(特殊普通合伙)11463代理人徐彦圣(51)Int.Cl.G06T5/00(2006.01)权利要求书2页说明书11页附图4页(54)发明名称人像美白方法、装置、电子设备和可读存储介质(57)摘要本申请实施例提供了一种人像美白方法、装置、电子设备和可读存储介质，涉及图像处理技术领域。该方法首先获取包含人像的待美白图像，然后将待美白图像输入人像美白模型进行美白，得到美白结果图像，其中，人像美白模型是将人像图像作为训练样本，对预先构建的包括人像美白主网络和人像掩膜次网络的人像处理网络进行训练，得到的训练后的人像美白主网络。如此，使用训练后的人像美白主网络对待美白图像中的人像进行美白的同时，保留了更多的图像细节，避免待美白图像因美白而失真。CN111784611ACN111784611A权利要求书1/2页1.一种人像美白方法，其特征在于，所述方法包括：获取包含人像的待美白图像；将所述待美白图像输入人像美白模型进行美白，得到美白结果图像，其中，所述人像美白模型是将人像图像作为训练样本，对预先构建的包括人像美白主网络和人像掩膜次网络的人像处理网络进行训练，得到的训练后的所述人像美白主网络。2.根据权利要求1所述的人像美白方法，其特征在于，所述人像美白模型是通过以下步骤训练得到的：获取人像图像和目标图像，其中，所述目标图像是通过对所述人像图像中的人像进行面部美白得到的；将所述人像图像作为训练样本、将所述目标图像作为标签，并采用预先构建的损失函数对所述人像处理网络进行训练，得到训练后的人像处理网络；将训练后的人像处理网络中的人像美白主网络作为所述人像美白模型。3.根据权利要求2所述的人像美白方法，其特征在于，所述目标图像包括目标掩膜图像和目标人像图像，所述人像美白主网络包括人像掩膜感知子网络和人像美白子网络；所述将所述人像图像作为训练样本、将所述目标图像作为标签，并采用预先构建的损失函数对所述人像处理网络进行训练，得到训练后的人像处理网络的步骤包括：将所述人像图像输入所述人像掩膜感知子网络，利用所述人像掩膜感知子网络对所述人像图像进行掩膜感知，得到掩膜感知图像；将所述掩膜感知图像输入所述人像掩膜次网络，利用所述人像掩膜次网络对所述掩膜感知图像进行人像掩膜处理，得到初步掩膜图像；将所述掩膜感知图像输入所述人像美白子网络，利用所述人像美白子网络对所述掩膜感知图像进行面部美白，得到美白后的初步结果图像；依据所述初步结果图像、所述初步掩膜图像、所述目标掩膜图像及所述目标人像图像，计算所述损失函数的损失值；依据所述损失值更新所述人像处理网络的参数，直至所述损失值满足预设条件，得到训练后的所述人像处理网络。4.根据权利要求3所述的人像美白方法，其特征在于，所述损失函数包括语义损失函数、L1损失函数及第一L2损失函数，所述损失值包括所述语义损失函数的第一输出值、所述L1损失函数的第二输出值及所述第一L2损失函数的第三输出值；所述依据所述初步结果图像、所述初步掩膜图像、所述目标掩膜图像及所述目标人像图像，计算所述损失函数的损失值的步骤包括：利用所述初步结果图像与所述目标人像图像，计算所述语义损失函数的第一输出值；利用所述初步结果图像与所述目标人像图像，计算所述L1损失函数的第二输出值；利用所述初步掩膜图像与所述目标掩膜图像，计算所述第一L2损失函数的第三输出值。5.根据权利要求4所述的人像美白方法，其特征在于，所述依据所述损失值更新所述人像处理网络的参数，直至所述损失值满足预设条件，得到训练后的所述人像处理网络的步骤包括：计算所述第一输出值、所述第二输出值及所述第三输出值的加权和；2CN111784611A权利要求书2/2页判断所述加权和是否小于预设阈值；若是，则停止更新所述人像处理网络的参数，得到训练后的所述人像处理网络；若否，则依据所述第一输出值、所述第二输出值及所述第三输出值更新所述人像处理网络的参数，并重复执行上述步骤，直至所述加权和小于所述预设阈值，得到训练后的所述人像处理网络。6.根据权利要求4所述的人像美白方法，其特征在于，所述语义损失函数包括预先训练的VGG模型及第二L2损失函数；所述利用所述初步结果图像与所述目标人像图像，计算所述语义损失函数的第一输出值的步骤包括：将所述初步结果图像输入所述VGG模型，得到第一功能图；将所述目标人像图像