[发明专利]发型迁移模型训练方法、发型迁移方法及相关装置

说明书

本申请提供发型迁移模型训练方法、发型迁移方法及相关装置，其中，发型迁移模型训练方法包括：获取样本发型图像和所样本参考发型图像；对样本发型图像和样本参考发型图像进行特征提取和特征融合，得到样本融合特征图像；对样本融合特征图像进行特征还原，以得到样本迁移发型图像；将样本迁移发型图像输入至图像判别模型和发型预测模型，以得到判别结果和发型预测结果；基于样本迁移发型图像和样本发型图像，以及，发型预测结果和样本参考发型图像中的发型，以及判别结果，对发型迁移模型、图像判别模型以及发型预测模型进行迭代调参，直至发型迁移模型、图像判别模型以及发型预测模型均收敛。该技术方案可以使虚拟美发的融合效果更自然。

技术领域

本申请涉及图像处理领域，尤其涉及发型迁移模型训练方法、发型迁移方法及相关装置。

背景技术

美容美发是在用户行为中一种目标较为常见的行为，在进行美容美发之前，许多用户希望看到自己美容美发后的形象，因此，一些基于图像处理技术的虚拟美发应用应运而生。

在目前的虚拟美发应用中，在获取到待美发图像后，通过分割算法先从待美发图像中分割出头发区域，然后将模板中的模板发型迁移至头发区域，即用美发模板中的模板发型替换头发区域中的发型，并将模板发型与待美发图像中除头发区域以外的其他区域进行融合，得到美发后的效果图。这种方式过度依赖模板，融合效果较生硬。

发明内容

本申请提供发型迁移模型训练方法、发型迁移方法及相关装置，以解决现有虚拟美发技术融合效果较生硬的问题。

第一方面，提供一种发型迁移模型训练方法，其中，发型迁移模型包括特征提取模块、特征融合模块以及特征还原模块，该方法包括如下步骤：

获取样本发型图像和上述样本发型图像对应的样本参考发型图像，其中，上述样本参考发型图像中的发型不同于上述样本发型图像中的发型；

通过上述特征提取模块分别对上述样本发型图像和上述样本参考发型图像进行特征提取，以得到第一样本特征图像和第二样本特征图像；

通过上述特征融合模块将上述第一样本特征图像与上述第二样本特征图像进行特征融合，得到样本融合特征图像；

通过上述特征还原模块对上述样本融合特征图像进行特征还原，以得到上述样本发型图像对应的样本迁移发型图像；

将上述样本迁移发型图像输入至图像判别模型，以得到上述样本迁移发型图像的判别结果，上述判别结果用于指示上述样本迁移发型图像为真实样本发型图像的可能性；

将上述样本迁移发型图像输入至发型预测模型，以得到上述样本迁移发型图像的发型预测结果；

基于上述样本迁移发型图像和上述样本发型图像，以及，上述发型预测结果和上述样本参考发型图像中的发型，以及上述判别结果，对上述发型迁移模型、上述图像判别模型以及上述发型预测模型进行迭代调参，直至上述发型迁移模型、上述图像判别模型以及上述发型预测模型均收敛；

将收敛后的发型迁移模型确定为目标发型迁移模型。