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

权利要求书

1.一种图像处理方法，其特征在于，包括：

获取训练样本集，其中，所述训练样本集包括源样本图像以及目标样本图像；

根据所述训练样本集训练图像转换模型和第一判别模型，所述源样本图像经过所述图像转换模型处理后得到目标图像；

根据所述第一判别模型计算所述目标图像和所述目标样本图像之间的所述第一对抗损失；

当所述第一对抗损失符合第一预设条件时，得到第一目标图像转换模型。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

当所述第一对抗损失不符合第一预设条件时，根据所述第一对抗损失对所述图像转换模型和第一判别模型中的参数进行调整；

所述第一对抗损失符合第一预设条件包括：所述第一对抗损失小于或等于第一预设阈值，或所述第一对抗损失不再变化。

3.根据权利要求1所述的方法，其特征在于，所述方法还包括：

根据所述训练样本集训练图像逆转换模型和第二判别模型，所述目标样本图像经过所述图像逆转换模型处理后得到第一源重构图像；

根据所述第二判别模型计算所述第一源重构图像和所述源样本图像的第二对抗损失；

基于所述第一对抗损失和第二对抗损失计算最终对抗损失；

当所述最终对抗损失不符合第二预设条件时，基于所述第一对抗损失和第二对抗损失对所述图像转换模型、图像逆转换模型、第一判别模型和第二判别模型中的参数进行调整，直至所述最终对抗损失符合所述第二预设条件，得到第二目标图像转换模型；

其中，所述最终对抗损失符合所述第二预设条件包括：所述最终对抗损失小于或等于第二预设阈值，或所述最终对抗损失不再变化。

4.根据权利要求3所述的方法，其特征在于，所述方法还包括：

根据所述训练样本集训练图像转换模型和图像逆转换模型，根据所述图像逆转换模型将所述目标图像转换为第二源重构图像；

根据所述图像转换模型将所述第一源重构图像转换为目标重构图像；

计算所述源样本图像与所述第二源重构图像之间的第一重构误差；

计算所述目标样本图像与所述目标重构图像之间的第二重构误差；

根据所述第一重构误差和第二重构误差计算最终重构误差；

当所述最终重构误差不符合第三预设条件时，基于所述第一重构误差和第二重构误差对所述图像转换模型和图像逆转换模型中的参数进行调整，直至所述最终重构误差符合所述第三预设条件，得到第三目标图像转换模型；

其中，所述最终重构误差符合所述第三预设条件包括：所述最终重构误差小于或等于第三预设阈值，或所述最终重构误差不再变化。

5.根据权利要求4所述的方法，其特征在于，所述方法还包括：

基于所述第一对抗损失、第二对抗损失、第一重构误差和第二重构误差计算最终损失；

当所述最终损失不符合第四预设条件时，基于所述第一对抗损失、第二对抗损失、第一重构误差和第二重构误差对所述图像转换模型和图像逆转换模型中的参数进行调整，直至所述最终损失符合所述第四预设条件，得到第四目标图像转换模型；

其中，所述最终损失符合所述第四预设条件包括：所述最终损失小于或等于第四预设阈值，或所述最终损失不再变化。

6.根据权利要求1-5任一项所述的方法，其特征在于，所述方法还包括：

获取源风格脸部图像；

根据图像转换模型将所述源风格脸部图像转换为目标风格脸部图像。

7.根据权利要求6所述的方法，其特征在于，所述根据图像转换模型将所述源风格脸部图像转换为目标风格脸部图像，包括：

提取所述源风格脸部图像的第一脸部特征信息；

将所述第一脸部特征信息转换为目标风格的第二脸部特征信息；

根据所述第二脸部特征信息生成所述目标风格脸部图像。