[发明专利]用户头像的生成方法及装置、电子设备、存储介质

说明书

本发明实施例提供了用户头像的生成方法及装置、电子设备、存储介质，所述方法包括：获取第一用户图像及第二用户图像；从所述第一用户图像中提取第一用户图像特征信息，以及从所述第二用户图像中提取第二用户图像特征信息；采用预先训练的生成模型，根据所述第一图像特征信息和所述第二图像特征信息生成目标用户图像。通过本发明实施例，实现了对用户头像的定制，使得用户头像兼容了个人特征和风格，保证了用户头像的个性化和专属性，且能够保障用户隐私。

技术领域

本发明涉及图像处理技术领域，特别是涉及用户头像的生成方法及装置、电子设备、存储介质。

背景技术

在注册账户后，通常需要设置账户对应的用户头像，其可以由用户自行上传，也可以直接采用默认的用户头像。

对于由用户上传的情况，上传的图像通常与用户个人相关，其可能包含隐私信息，用户不想在网络上公开，但又缺乏修图能力，进而可能直接采用默认的用户头像，而默认的用户头像又难以突出用户的个性特点。

发明内容

鉴于上述问题，提出了本发明以便提供克服上述问题或者至少部分地解决上述问题的用户头像的生成方法及装置、电子设备、存储介质，包括：

一种用户头像的生成方法，所述方法包括：

获取第一用户图像及第二用户图像；

从所述第一用户图像中提取第一用户图像特征信息，以及从所述第二用户图像中提取第二用户图像特征信息；

采用预先训练的生成模型，根据所述第一图像特征信息和所述第二图像特征信息生成目标用户图像。

可选地，还包括：

获取第一样本图像及第二样本图像；其中，所述第一样本图像与所述第一用户图像相关联，所述第二样本图像与所述第二用户图像相关联；

对所述第一样本图像及所述第二样本图像进行训练，得到所述生成模型。

可选地，对所述第一样本图像及所述第二样本图像进行训练，得到所述生成模型，包括：

生成所述第一样本图像及所述第二样本图像对应的第三样本图像；

获取所述第二样本图像对应的第二样本特征信息，以及获取所述第三样本图像对应的第三样本特征信息；

采用所述第二样本特征信息信息和所述第三样本特征信息信息，计算得到纹理损失；

根据预先训练的判别模型和所述纹理损失，更新所述生成模型。

可选地，根据预先训练的判别模型和所述纹理损失，更新所述生成模型，包括：

采用所述预先训练的所述判别模型，确定所述第三样本图像对应的内容损失；

根据所述内容损失和所述纹理损失的权值，更新所述生成模型。

可选地，还包括：

获取第四样本图像、第五样本图像，以及所述第四样本图像及所述第五样本图像对应的第六样本图像；其中，所述第四样本图像与所述第一用户图像相关联，所述第五样本图像与所述第二用户图像相关联；

将所述第四样本图像和所述第六样本图像组织为正样本，以及将所述第五样本图像和所述第六样本图像组织为负样本；

分别对所述正样本和所述负样本进行训练，得到判别模型；其中，所述判别模型对应的正样本损失在预设的正样本损失范围内，所述判别模型对应的负样本损失在预设的负样本损失范围内。

可选地，采用如下方式计算所述判别模型对应的正样本损失：

确定所述正样本对应的正样本矩阵；