皮肤区域提取模型的训练方法、直播图像处理方法及装置制造方法及图纸

本申请涉及图像处理及直播技术领域，提供一种皮肤区域提取模型的训练方法、处理直播图像中皮肤区域的方法、装置、设备和介质。本申请能够训练获得可准确稳定提取皮肤区域的模型。方法包括：对样本图像进行针对图像场景的增强处理得到增强样本图像及其标注图像，对增强样本图像进行针对皮肤区域亮度的第一、第二亮度增强处理得到第一、第二模型输入图像，将第一、第二模型输入图像输入待训练的皮肤区域提取模型获得第一、第二模型输出图像，获取表征第一、第二模型输出图像的一致性的第一模型损失、表征第一、第二模型输出图像各自与标注图像的一致性的第二、第三模型损失，根据第一、第二和第三模型损失对该模型进行训练。二和第三模型损失对该模型进行训练。二和第三模型损失对该模型进行训练。

【技术实现步骤摘要】
皮肤区域提取模型的训练方法、直播图像处理方法及装置


[0001]本申请涉及图像处理及直播
，特别是涉及一种皮肤区域提取模型的训练方法、处理直播图像中皮肤区域的方法、装置、电子设备和计算机可读存储介质。

技术介绍

[0002]随着人工智能技术的发展，基于深度学习的图像处理被应用于各种领域以执行相应的图像处理任务，其中包含利用基于深度学习的皮肤区域处理模型应用于对图像中皮肤区域的处理任务中。然而，在目前的技术中，此类皮肤区域处理模型多集中于对如皮肤病理、病变或皮肤分析中，这些模型对皮肤区域的提取容易受到复杂场景的影响，其对皮肤区域的提取稳定性和准确性较低。

技术实现思路

[0003]基于此，有必要针对上述技术问题，提供一种皮肤区域提取模型的训练方法、处理直播图像中皮肤区域的方法、装置、电子设备和计算机可读存储介质。
[0004]第一方面，本申请提供了一种皮肤区域提取模型的训练方法。所述方法包括：
[0005]获取包含皮肤区域的样本图像集及用于标注所述皮肤区域的标注图像集；
[0006]针对所述样本图像集中的样本图像，基于所述样本图像集及标注图像集，对所述样本图像进行针对图像场景的增强处理，得到增强样本图像及其标注图像；
[0007]基于所述增强样本图像及其标注图像，对所述增强样本图像进行针对皮肤区域亮度的第一亮度增强处理和第二亮度增强处理，得到第一模型输入图像和第二模型输入图像；
[0008]将所述第一模型输入图像和第二模型输入图像输入待训练的皮肤区域提取模型，获得所述待训练的皮肤区域提取模型输出的所述第一模型输入图像对应的第一模型输出图像和所述第二模型输入图像对应的第二模型输出图像；
[0009]获取表征所述第一模型输出图像和第二模型输出图像的一致性的第一模型损失，获取表征所述第一模型输出图像和第二模型输出图像各自与标注图像的一致性的第二模型损失和第三模型损失；
[0010]根据所述第一模型损失、第二模型损失和第三模型损失，对所述待训练的皮肤区域提取模型进行训练。
[0011]在其中一个实施例中，所述对所述样本图像进行针对图像场景的增强处理，包括：从所述样本图像集中获取所述样本图像的辅助图像；将所述样本图像进行尺寸放大处理，在尺寸放大的样本图像中划分辅助图像容纳区域；将所述辅助图像对应放入所述尺寸放大的样本图像中的辅助图像容纳区域中，将放入辅助图像的样本图像进行尺寸恢复处理。
[0012]在其中一个实施例中，所述对所述样本图像进行针对图像场景的增强处理，包括：从所述样本图像集中获取所述样本图像的辅助图像；按照一定概率将所述样本图像和辅助图像进行缩放处理后，将缩放处理后的辅助图像中的皮肤区域添加到缩放处理后的样本图
像中。
[0013]在其中一个实施例中，所述对所述样本图像进行针对图像场景的增强处理，包括：
[0014]对所述样本图像的标注图像进行腐蚀处理；根据腐蚀处理后的标注图像与腐蚀处理前的标注图像，确定所述样本图像的边缘皮肤区域；对所述样本图像的标注图像进行膨胀处理；根据膨胀处理后的标注图像与膨胀处理前的标注图像，确定所述样本图像的待填充边缘区域；根据所述边缘皮肤区域的像素值对所述样本图像的待填充边缘区域进行填充处理。
[0015]在其中一个实施例中，所述对所述样本图像进行针对图像场景的增强处理，包括：根据所述样本图像的标注图像，确定所述样本图像的皮肤区域；根据所述样本图像的皮肤区域的像素值统计信息，获得类肤色噪声信息；将所述类肤色噪声信息添加至所述样本图像中。
[0016]在其中一个实施例中，所述第一亮度增强处理或第二亮度增强处理，包括：根据所述增强样本图像的标注图像，确定所述增强样本图像的边缘皮肤区域；根据所述增强样本图像的边缘皮肤区域确定待调整图像块；对所述待调整图像块进行亮度增强处理。
[0017]在其中一个实施例中，所述第一亮度增强处理或第二亮度增强处理，包括：根据所述增强样本图像的标注图像，确定所述增强样本图像的边缘皮肤区域；对所述边缘皮肤区域进行亮度增强处理。
[0018]第二方面，本申请提供了一种处理直播图像中皮肤区域的方法。应用于终端。所述方法包括：
[0019]获取经训练的皮肤区域提取模型；所述皮肤区域提取模型根据如上所述的方法训练得到；
[0020]获得待处理的直播图像后，将所述直播图像输入所述皮肤区域提取模型，根据所述皮肤区域提取模型输出的皮肤区域预测图像，确定所述直播图像中的皮肤区域；
[0021]在所述直播图像中的皮肤区域添加皮肤处理效果。
[0022]第三方面，本申请提供了一种皮肤区域提取模型的训练装置。所述装置包括：
[0023]图像集获取模块，用于获取包含皮肤区域的样本图像集及用于标注所述皮肤区域的标注图像集；
[0024]第一增强处理模块，用于针对所述样本图像集中的样本图像，基于所述样本图像集及标注图像集，对所述样本图像进行针对图像场景的增强处理，得到增强样本图像及其标注图像；
[0025]第二增强处理模块，用于基于所述增强样本图像及其标注图像，对所述增强样本图像进行针对皮肤区域亮度的第一亮度增强处理和第二亮度增强处理，得到第一模型输入图像和第二模型输入图像；
[0026]图像输入模块，用于将所述第一模型输入图像和第二模型输入图像输入待训练的皮肤区域提取模型，获得所述待训练的皮肤区域提取模型输出的所述第一模型输入图像对应的第一模型输出图像和所述第二模型输入图像对应的第二模型输出图像；
[0027]损失获取模块，用于获取表征所述第一模型输出图像和第二模型输出图像的一致性的第一模型损失，获取表征所述第一模型输出图像和第二模型输出图像各自与标注图像的一致性的第二模型损失和第三模型损失；
[0028]训练处理模块，用于根据所述第一模型损失、第二模型损失和第三模型损失，对所述待训练的皮肤区域提取模型进行训练。
[0029]第四方面，本申请提供了一种处理直播图像中皮肤区域的装置。应用于终端。所述装置包括：
[0030]模型获取模块，用于获取经训练的皮肤区域提取模型；所述皮肤区域提取模型根据如上所述的方法训练得到；
[0031]区域确定模块，用于获得待处理的直播图像后，将所述直播图像输入所述皮肤区域提取模型，根据所述皮肤区域提取模型输出的皮肤区域预测图像，确定所述直播图像中的皮肤区域；
[0032]效果添加模块，用于在所述直播图像中的皮肤区域添加皮肤处理效果。
[0033]第五方面，本申请提供了一种电子设备。所述电子设备包括存储器和处理器，所述存储器存储有计算机程序，所述处理器执行所述计算机程序时实现以下步骤：
[0034]获取包含皮肤区域的样本图像集及用于标注所述皮肤区域的标注图像集；针对所述样本图像集中的样本图像，基于所述样本图像集及标注图像集，对所述样本图像进行针对图像场景的增强处理，得到增强样本图像及其标注图像；基于所述增强样本图像及其标注图像本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种皮肤区域提取模型的训练方法，其特征在于，所述方法包括：获取包含皮肤区域的样本图像集及用于标注所述皮肤区域的标注图像集；针对所述样本图像集中的样本图像，基于所述样本图像集及标注图像集，对所述样本图像进行针对图像场景的增强处理，得到增强样本图像及其标注图像；基于所述增强样本图像及其标注图像，对所述增强样本图像进行针对皮肤区域亮度的第一亮度增强处理和第二亮度增强处理，得到第一模型输入图像和第二模型输入图像；将所述第一模型输入图像和第二模型输入图像输入待训练的皮肤区域提取模型，获得所述待训练的皮肤区域提取模型输出的所述第一模型输入图像对应的第一模型输出图像和所述第二模型输入图像对应的第二模型输出图像；获取表征所述第一模型输出图像和第二模型输出图像的一致性的第一模型损失，获取表征所述第一模型输出图像和第二模型输出图像各自与标注图像的一致性的第二模型损失和第三模型损失；根据所述第一模型损失、第二模型损失和第三模型损失，对所述待训练的皮肤区域提取模型进行训练。2.根据权利要求1所述的方法，其特征在于，所述对所述样本图像进行针对图像场景的增强处理，包括：从所述样本图像集中获取所述样本图像的辅助图像；将所述样本图像进行尺寸放大处理，在尺寸放大的样本图像中划分辅助图像容纳区域；将所述辅助图像对应放入所述尺寸放大的样本图像中的辅助图像容纳区域中，将放入辅助图像的样本图像进行尺寸恢复处理。3.根据权利要求1所述的方法，其特征在于，所述对所述样本图像进行针对图像场景的增强处理，包括：从所述样本图像集中获取所述样本图像的辅助图像；按照一定概率将所述样本图像和辅助图像进行缩放处理后，将缩放处理后的辅助图像中的皮肤区域添加到缩放处理后的样本图像中。4.根据权利要求1所述的方法，其特征在于，所述对所述样本图像进行针对图像场景的增强处理，包括：对所述样本图像的标注图像进行腐蚀处理；根据腐蚀处理后的标注图像与腐蚀处理前的标注图像，确定所述样本图像的边缘皮肤区域；对所述样本图像的标注图像进行膨胀处理；根据膨胀处理后的标注图像与膨胀处理前的标注图像，确定所述样本图像的待填充边缘区域；根据所述边缘皮肤区域的像素值对所述样本图像的待填充边缘区域进行填充处理。5.根据权利要求1所述的方法，其特征在于，所述对所述样本图像进行针对图像场景的增强处理，包括：根据所述样本图像的标注图像，确定所述样本图像的皮肤区域；根据所述样本图像的皮肤区域的像素值统计信息，获得类肤色噪声信息；
将所述类肤色噪声信息添加至所述样本图像中。6.根据权利要求1至5任一项所述的方法，其特征在于，所述第一亮度增强处理或第二亮度增强处理，包括：根据所述增强样本图像的标注图像，确定所述增强样本图像的边缘皮肤区域；根据所述增强样本图像的边缘皮肤区域确定待调整图像块；对...

【专利技术属性】
技术研发人员：郑康元，钟耿君，陈增海，
申请(专利权)人：广州方硅信息技术有限公司，
类型：发明
国别省市：