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

说明书

本公开实施例提供了一种图像处理方法、装置、电子设备以及存储介质。该方法包括：获取包括第一发型信息的待处理图像，并确定与待处理图像的头发区域对应的目标掩膜图像；将待处理图像与目标掩膜图像进行拼接，得到目标处理图像；将目标处理图像输入至预先训练完成的发型更换模型中，得到与待处理图像对应的包含第二发型信息的目标展示图像；其中，发型更换模型基于包括头发区域的样本原始图像、与样本原始图像的头发区域对应的样本掩膜图像、与样本原始图像对应的期望图像以及与期望图像的头发区域对应的期望掩膜图像训练得到。通过本公开实施例公开的技术方案，真实展示用户更换发型后的发型效果，从而提高了用户对于发型预览的效果体验。

技术领域

本公开实施例涉及图像处理技术，尤其涉及一种图像处理方法、装置、电子设备以及存储介质。

背景技术

在多媒体技术日益发展的今日，能够自动识别并且更换发型的方法是人们的迫切需求，计算机视觉技术使得这种需求得以实现。通过计算机视觉模型，能够自动识别、分析、修改人物图片中的发型因素，有很大的实际需求。用户可以根据自己的喜好，通过计算机模型尝试更换不同的发型，比较效果后，再去寻找理发师达到最令人满意的发型效果。

现有技术中基于计算机模型对发型进行更换的方法与人脸的贴合度较差，无法较好地满足用户对于更换后的发型进行真实效果预览的需求。

发明内容

本公开提供一种发明名称，以实现真实展示用户更换发型后的发型效果，从而提高了用户对于发型预览的效果体验。

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

获取包括第一发型信息的待处理图像，并确定与所述待处理图像的头发区域对应的目标掩膜图像；

将所述待处理图像与所述目标掩膜图像进行拼接，得到目标处理图像；

将所述目标处理图像输入至预先训练完成的发型更换模型中，得到与所述待处理图像对应的包含第二发型信息的目标展示图像；

其中，所述发型更换模型基于包括头发区域的样本原始图像、与所述样本原始图像的头发区域对应的样本掩膜图像、与所述样本原始图像对应的期望图像以及与所述期望图像的头发区域对应的期望掩膜图像训练得到。

第二方面，本公开实施例还提供了一种图像处理装置，该装置包括：

图像获取模块，用于获取包括第一发型信息的待处理图像，并确定与所述待处理图像的头发区域对应的目标掩膜图像；

图像拼接模块，用于将所述待处理图像与所述目标掩膜图像进行拼接，得到目标处理图像；

发型更换模块，用于将所述目标处理图像输入至预先训练完成的发型更换模型中，得到与所述待处理图像对应的包含第二发型信息的目标展示图像；

其中，所述发型更换模型基于包括头发区域的样本原始图像、与所述样本原始图像的头发区域对应的样本掩膜图像、与所述样本原始图像对应的期望图像以及与所述期望图像的头发区域对应的期望掩膜图像训练得到。

第三方面，本公开实施例还提供了一种电子设备，所述电子设备包括：

一个或多个处理器；

存储装置，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如本发明任意实施例提供的图像处理方法。

第四方面，本公开实施例还提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现本发明任意实施例提供的图像处理方法。