[发明专利]图像处理方法、装置及电子设备

说明书

本公开关于一种图像处理方法、装置及电子设备。所述方法包括：将待处理图像输入到人物分割模块中进行计算，得到所述人物分割模块输出的所述待处理图像中包含的人物图像；将所述待处理图像输入到头发分割模块中进行计算，得到所述头发分割模块输出的所述待处理图像中包含的精细化效果的头发图像；将所述人物图像与所述精细化效果的头发图像进行融合，得到包含精细头发效果的人物图像。应用本方法，可以从待处理图像中分割出头发边界清晰、发丝边缘锐利的人物图像，实现了发丝级的精细头发分割的效果。

技术领域

本公开涉及互联网技术领域，尤其涉及一种图像处理方法、装置及电子设备。

背景技术

人物分割技术是一种特定类别的语义分割技术。在实现时，需要对图像中的每个像素点进行二分类，以确定每个像素点的属性类别，该属性类别一般包括人物属性和背景属性。进而基于属性类别，将属于人物属性的像素点提取出来构建出仅包含人物像素的图像。如此，实现从图像中分割出人物图像的目的。

在现有的是人物分割技术中，通常可以采用人物分割模块进行人物分割。然而，由于现有的人物分割模块的学习能力和样本标注精细度的限制，使用人物分割模块虽然可以获得人物图像，但是这样的人物图像中存在头发区域的边界不够清晰，发丝边缘不够锐利，有较强的羽化效果，无法分割出具有发丝级的精细头发效果的人物图像的问题。

发明内容

本公开提供一种图像处理方法、装置及电子设备，以至少解决相关技术中无法从原始图像中分割出具有精细发丝效果的人物图像的问题。本公开的技术方案如下：

根据本公开实施例的第一方面，提供一种图像处理方法，包括：

将待处理图像输入到人物分割模块中进行计算，得到所述人物分割模块输出的所述待处理图像中包含的人物图像；

将所述待处理图像输入到头发分割模块中进行计算，得到所述头发分割模块输出的所述待处理图像中包含的精细化效果的头发图像；

将所述人物图像与所述精细化效果的头发图像进行融合，得到包含精细头发效果的人物图像。

可选的，所述方法还包括：

执行以下步骤，对所述人物分割模块和所述头发分割模块进行多任务学习训练，直到所述多任务学习训练的损失函数的损失值达到最优，其中，用于训练的图像样本集中的样本图像被标注了人物所在区域和头发所在区域：

将所述图像样本集中的样本图像输入到所述人物分割模块，以所述样本图像中被标注的人物所在区域为监督信号，计算从所述样本图像中分割出人物图像时，所述人物分割模块的第一损失函数的第一损失值；

将所述人物分割模块输出的人物图像输入到所述头发分割模块，以所述样本图像中被标注的头发所在区域为监督信号，计算从所述人物图像中分割出的精细头发效果时，所述头发分割模块的第二损失函数的第二损失值；

将所述第一损失值和第二损失值进行加权计算，得到所述多任务学习训练的损失函数的损失值，并基于所述损失值分别调整所述人物分割模块和所述头发分割模块的模块参数。

可选的，所述计算从所述人物图像中分割出的精细头发效果时，所述头发分割模块的第二损失函数的第二损失值，包括：

将所述头发分割模块从所述人物分割模块输出的人物图像中分割出的精细头发效果，和与所述样本图像对应的人物边缘区域进行计算，得到所述样本图像的头发边缘区域；

将所述头发边缘区域和所述人物分割模块输出的人物图像进行计算，得到所述头发分割模块的第二损失函数的第二损失值。

可选的，所述将所述头发分割模块从所述人物分割模块输出的人物图像中分割出的精细头发效果，和与所述样本图像对应的人物边缘区域进行计算，得到所述样本图像的头发边缘区域，包括：