头发变换模型生成方法和装置、头发变换方法和装置制造方法及图纸

本公开关于一种头发变换模型生成方法和装置、头发变换方法和装置，涉及图像处理领域，能够实现人物头像的头发变换后和变换前相比颜色更一致自然的头发变换功能。该模型生成方法包括：获取第一样本人物图像和与第一样本人物图像对应的第二样本人物图像；第二样本人物图像由第一样本人物图像得到且发型与第一样本人物图像不一致；根据第一样本人物图像和第二样本人物图像确定第一颜色引导图；第一颜色引导图中包括与第二样本人物图像中头发分割区域相同的头发生成区域，且其头发生成区域的颜色是根据第一样本人物图像中头发的颜色得到的；以第一样本人物图像和第一颜色引导图作为训练数据，第二样本人物图像作为监督信息，训练得到头发变换模型。训练得到头发变换模型。训练得到头发变换模型。

【技术实现步骤摘要】
头发变换模型生成方法和装置、头发变换方法和装置


[0001]本公开实施例涉及图像处理领域，尤其涉及一种头发变换模型生成方法和装置、头发变换方法和装置。

技术介绍

[0002]目前，为了提高用户对视频应用的兴趣，大多数视频应用中都具备有头发变换功能，即将用户提供的照片或视频中用户的头发变为另一种发型。现有技术中对于这一功能的实现具体包括以下两个步骤：(1)利用预设的发型转换模型将用户提供的照片或视频中的人物的原发型转换为预设发型(例如大波浪长发)。(2)根据原发型的颜色利用预设颜色变换方法，将(1)步骤转换后的预设发型的颜色变换为贴近用户提供的照片或视频中人物的头发颜色。其中，(2)步骤中目前会使用两种方式：查表和均值方差颜色变换。其中，查表(Lookup Table)的需要基于设计人员的经验进行，难以适用当前用户发色各异的复杂应用场景，头发变换后的颜色和变换前的颜色一致性不好。而均值方差颜色变换在实现过程易受原发型中颜色对应的RGB值中的异常值影响(例如外界高光对颜色的影响)导致变换后的预设发型的颜色不自然。
[0003]综上，现有技术在实现头发变换功能时中，变换前和变换后的头发颜色一致性较差，不够自然。

技术实现思路

[0004]本公开关于一种头发变换模型生成方法和装置、头发变换方法和装置，能够实现人物头像的头发变换后和变换前相比颜色更一致自然的头发变换功能。
[0005]为达到上述目的，本公开实施例采用如下技术方案：
[0006]第一方面，提供一种头发变换模型生成方法，包括：获取第一样本人物图像和与第一样本人物图像对应的第二样本人物图像；第二样本人物图像是根据第一样本人物图像得到的，第二样本人物图像中人物的发型与第一样本人物图像不一致；根据第一样本人物图像和第二样本人物图像确定第一颜色引导图；第一颜色引导图中包括与第二样本人物图像中头发分割区域相同的头发生成区域；第一颜色引导图中的头发生成区域的颜色是根据第一样本人物图像中头发的颜色得到的；以第一样本人物图像和第一颜色引导图作为训练数据，第二样本人物图像作为监督信息，训练得到头发变换模型。
[0007]基于上述技术方案，在训练头发变换模型时，首先获取多对样本对，每一对样本对中的第一样本人物图像和第二样本人物图像的发型不一致，且第二样本人物图像是由第一样本人物图像得到的；然后根据这些样本对，生成对应的第一颜色引导图，第一颜色引导图中的头发生成区域与第二样本人物图像中的头发分割区域相同，且其颜色是由第一样本人物图像中头发的颜色得到的。最后以第一样本人物图像和第一颜色引导图作为训练数据，第二样本人物图像作为监督信息，便可以训练得到头发变换模型。这样一来，得到的头发变换模型，在接收到一个人物图像后以及指示相应发型的信息后，便可以输出头发变换后的
人物图像。因为整个模型训练使用的样本是大量的，并使用颜色由第一样本人物图像决定的第一颜色引导图，同时使用与第一颜色引导图中的发型相同的第二样本人物图像作为监督信息，所以最终得到的模型，可以使得其在完成头发变换功能时，变换后的头发和变换前的头发颜色一致性较好。
[0008]可选的，根据第一样本人物图像和第二样本人物图像确定第一颜色引导图包括：
[0009]确定第二样本人物图像的目标头发掩码图；根据第一样本人物图像中头发的颜色值、目标头发掩码图和目标头发掩码图对应的头发的颜色值确定第一颜色引导图。
[0010]进一步可选的，根据第一样本人物图像中头发的颜色值、目标头发掩码图和目标头发掩码图对应的头发的颜色值确定第一颜色引导图包括：
[0011]计算第一样本人物图像中头发的颜色值的统计值；将目标头发掩码图中头发的所有像素的颜色值替换为第一样本人物图像中头发的颜色值的统计值，以得到第一颜色引导图。
[0012]可选的，在第二样本人物图像中头发的区域大于第一样本人物图像中头发的区域的情况下，第一颜色引导图中的头发生成区域的颜色是根据第一样本人物图像中头发的颜色得到的包括：第一颜色引导图中的头发生成区域的颜色是根据第一样本人物图像中头发的颜色和第二样本人物图像中头发的颜色得到的；
[0013]根据第一样本人物图像中头发的颜色值、目标头发掩码图和目标头发掩码图对应的头发的颜色值确定第一颜色引导图包括：
[0014]计算第一样本人物图像中头发的颜色值的统计值，以及目标头发掩码图中除对应第一样本人物图像中头发的区域的第一区域以外的第二区域对应的头发的颜色值的统计值；
[0015]将第一区域中所有像素的颜色值替换为第一样本人物图像中头发的颜色值的统计值，并将第二区域中所有像素的颜色值替换为第二区域对应的头发的颜色值的统计值，以得到第一颜色引导图。
[0016]可选的，在第一样本人物图像中头发的区域大于第二样本人物图像中头发的区域的情况下，根据第一样本人物图像中头发的颜色值、目标头发掩码图和目标头发掩码图对应的头发的颜色值确定第一颜色引导图包括：
[0017]计算第一样本人物图像中对应目标头发掩码图中头发的区域的第三区域中头发的颜色值的统计值；将目标头发掩码图中头发的区域中所有像素的颜色值替换为第三区域中头发颜色值的统计值，以得到第一颜色引导图。
[0018]可选的，以第一样本人物图像和颜色引导图作为训练数据，第二样本人物图像作为监督信息，训练得到头发变换模型之后还包括：
[0019]获取多个第三样本人物图像；获取至少一个预设颜色引导图；每一个预设颜色引导图对应一种头发生成区域；计算第三样本人物图像中头发的颜色值的统计值；将预设颜色引导图中头发的所有像素的颜色值替换为第三样本人物图像中头发的颜色值的统计值，以得到第二颜色引导图；将第三样本人物图像和第二颜色引导图输入头发变换模型，得到第四样本人物图像；以第三样本人物图像和发型变量为训练数据，发型变量对应的第四样本人物图像为监督信息，训练得到目标头发变换模型；发型变量与头发生成区域一一对应；将头发变换模型更新为目标头发变换模型。
[0020]可选的，计算头发的颜色值的统计值包括：将头发的颜色的所有像素的RGB值按照颜色和值升序排列后，计算次序在预设次序范围内的颜色和值对应的RGB值的平均值，并将次序在预设次序范围内的颜色和值对应的RGB值的平均值作为头发的颜色值的统计值；颜色和值为RGB值中R值、G值和B值的和。
[0021]第二方面，提供一种头发变换方法，包括：
[0022]获取源人物图像和发型指示信息；发型指示信息用于指示目标头发生成区域；将源人物图像和发型指示信息输入头发变换模型中，得到目标人物图像；头发变换模型由如第一方面提供的头发变换模型生成方法得到；目标人物图像中人物的发型为发型指示信息指示的目标头发生成区域对应的发型，且目标人物图像中的人物图像的头发的颜色与源人物图像中的人物图像的头发的颜色的相似度大于预设值。
[0023]基于上述技术方案，需要实现头发变换功能的头发变换装置，可以只使用第一方面提供的技术方案生成的头发变换模型便能利用源人物图本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种头发变换模型生成方法，其特征在于，包括：获取第一样本人物图像和与所述第一样本人物图像对应的第二样本人物图像；所述第二样本人物图像是根据所述第一样本人物图像得到的，所述第二样本人物图像中人物的发型与所述第一样本人物图像不一致；根据所述第一样本人物图像和所述第二样本人物图像确定第一颜色引导图；所述第一颜色引导图中包括与所述第二样本人物图像中头发分割区域相同的头发生成区域；所述第一颜色引导图中的头发生成区域的颜色是根据所述第一样本人物图像中头发的颜色得到的；以所述第一样本人物图像和所述第一颜色引导图作为训练数据，所述第二样本人物图像作为监督信息，训练得到头发变换模型。2.根据权利要求1所述的头发变换模型生成方法，其特征在于，所述根据所述第一样本人物图像和所述第二样本人物图像确定第一颜色引导图包括：确定所述第二样本人物图像的目标头发掩码图；根据所述第一样本人物图像中头发的颜色值、所述目标头发掩码图和所述目标头发掩码图对应的头发的颜色值确定所述第一颜色引导图。3.根据权利要求2所述的头发变换模型生成方法，其特征在于，所述根据所述第一样本人物图像中头发的颜色值、所述目标头发掩码图和所述目标头发掩码图对应的头发的颜色值确定所述第一颜色引导图包括：计算所述第一样本人物图像中头发的颜色值的统计值；将所述目标头发掩码图中头发的所有像素的颜色值替换为所述第一样本人物图像中头发的颜色值的统计值，以得到所述第一颜色引导图。4.根据权利要求2所述的头发变换模型生成方法，其特征在于，在所述第二样本人物图像中头发的区域大于所述第一样本人物图像中头发的区域的情况下，所述第一颜色引导图中的头发生成区域的颜色是根据所述第一样本人物图像中头发的颜色得到的包括：所述第一颜色引导图中的头发生成区域的颜色是根据所述第一样本人物图像中头发的颜色和第二样本人物图像中头发的颜色得到的；所述根据所述第一样本人物图像中头发的颜色值、所述目标头发掩码图和所述目标头发掩码图对应的头发的颜色值确定所述第一颜色引导图包括：计算所述第一样本人物图像中头发的颜色值的统计值，以及所述目标头发掩码图中除对应所述第一样本人物图像中头发的区域的第一区域以外的第二区域对应的头发的颜色值的统计值；将所述第一区域中所有像素的颜色值替换为所述第一样本人物图像中头发的颜色值的统计值，并将所述第二区域中所有像素的颜色值替换为所述第二区域对应的头发的颜色值的统计值，以得到所述第一颜色引导图。5.根据权利要求2所述的头发变换模型生成方法，其特征在于，在所述第一样本人物图像中头发的区域大于所述第二样本人物图像中头发的区域的情况下，所述根据所述第一样本人物图像中头发的颜色值、所述目标头发掩码图和所述目标头发掩码图对应的头发的颜色值确定所述第一颜色引导图包括：计算所述第一样本人物图像中对应所述目标头发掩码图中头发的区域的第三区域中
头发的颜色值的统计值；将所述目标头发掩码图中头发的区域中所有像素的颜色值替换为所述第三区域中头发颜色值的统计值，以得到所述第一颜色引导图。6.根据权利要求1所述的头发变换模型生成方法，其特征在于，以所述第一样本人物图像和所述颜色引导图作为训练数据，所...

【专利技术属性】
技术研发人员：高远，黄慧娟，孙阳，宋丛礼，郑文，
申请(专利权)人：北京达佳互联信息技术有限公司，
类型：发明
国别省市：