[发明专利]三维发型匹配方法及电子设备

权利要求书

1.一种电子设备，其特征在于，包括处理器和显示单元；

其中，所述处理器，被配置为：

对包含用户的目标图像进行头发检测处理，得到中间图像，其中所述中间图像为包含头发所在区域的图像；

利用预设的头发分割算法，对所述中间图像进行头发分割，得到包含所述头发的前景图像；

将所述前景图像输入至预先训练好的发型匹配网络中进行识别，得到与所述前景图像包含的头发匹配的三维发型；

所述显示单元，被配置为显示所述三维发型。

2.根据权利要求1所述的电子设备，其特征在于，所述处理器，还被配置为：

所述得到与所述前景图像包含的头发匹配的三维发型之后，提取所述前景图像中的头发颜色，利用所述头发颜色对所述三维发型的颜色进行填充，得到目标三维发型。

3.根据权利要求2所述的电子设备，其特征在于，所述处理器，还被配置为：

所述得到目标三维发型之后，利用预设的目标三维发型与三维发型渲染调节方式的对应关系，确定与所述目标三维发型相对应的三维发型渲染调节方式；其中所述三维发型渲染调节方式包括基础色调节、散射调节、高光调节、切线调节、背光度调节、环境光遮蔽调节以及深度偏移调节中的至少一个；

利用确定的所述三维发型渲染调节方式对所述目标三维发型进行渲染，得到渲染后的目标三维发型。

4.根据权利要求1所述的电子设备，其特征在于，所述处理器在执行所述对包含用户的目标图像进行头发检测处理，得到中间图像，具体被配置为：

将所述目标图像均分成指定数量的图像块；

按照预设的顺序分别对各图像块进行头发检测处理，得到所述中间图像；

其中，针对任一图像块进行头发检测处理的方法为：

对所述图像块进行特征提取，得到所述图像块的方向梯度直方图特征和局部三值模式特征；并，

将所述方向梯度直方图特征和局部三值模式特征输入至预先训练好的随机森林模型中，以得到所述图像块的分类结果；

若所述图像块的分类结果为不包含头发的类型，则将所述图像块的像素值设置为指定像素值。

5.根据权利要求1所述的电子设备，其特征在于，所述处理器在执行所述将所述前景图像输入至预先训练好的发型匹配网络中进行识别，得到与所述前景图像包含的头发匹配的三维发型，具体被配置为：

利用发型匹配网络对所述前景图像进行特征提取，得到第一发型特征信息；

将所述第一发型特征信息与各三维发型对应的二维发型图像的第二发型特征信息进行比对，得到所述第一发型特征信息与各第二发型特征信息的相似度；

将与所述第一发型特征信息相似度最高的第二发型特征信息对应的三维发型确定为与所述前景图像包含的头发匹配的三维发型。

6.根据权利要求1所述的电子设备，其特征在于，所述处理器，还被配置为：

通过以下方式训练所述发型匹配网络：

获取各三维发型训练样本，其中任意一个三维发型训练样本中包括同一发型的前景图像以及与所述前景图像相对应的标注三维发型，且不同三维发型训练样本中前景图像的发型不相同；

针对任意一个三维发型训练样本执行以下步骤：

将所述三维发型训练样本输入至发型匹配网络中，对所述三维训练样本中的前景图像进行特征提取，得到第一发型特征信息；并，

将所述第一发型特征信息与各三维发型对应的二维图像的第二发型特征信息进行比对，得到所述第一发型特征信息与各第二发型特征信息的相似度；

确定与所述第一发型特征信息相似度最高的第二发型特征信息所对应的三维发型；

将确定的所述三维发型与所述标注三维发型进行比对，得到误差值；

若所述误差值未满足指定条件，则对所述发型匹配网络的训练参数进行调节后，返回执行将所述三维发型训练样本输入至发型匹配网络中，对所述三维训练样本中的前景图像进行特征提取的步骤，直至所述误差值满足所述指定条件，结束对所述发型匹配网络的训练。