[发明专利]图像转换模型的生成方法、装置、电子设备及存储介质

权利要求书

1.一种图像转换模型的生成方法，其特征在于，所述方法包括：

获取第一样本图像集，所述第一样本图像集包括多个真人图像和多个卡通图像，每个所述卡通图像的分类类别属于第一分类类别或第二分类类别，属于所述第一分类类别的卡通图像的第一数量少于属于所述第二分类类别的卡通图像的第二数量，属于所述第二分类类别的卡通图像的卡通风格一致，其中，分类类别是根据图像中的人物特征确定的；

基于所述第一样本图像集训练第一初始模型，直至达到第一停止训练条件，得到第二初始模型，所述第一初始模型用于计算图像的注意力特征图，并基于所述注意力特征图设置归一化网络层的网络参数，基于完成网络参数设置的归一化网络层对所述注意力特征图进行解码，得到转换后的图像；

将属于所述第一分类类别的真人图像集输入至所述第二初始模型中，得到属于所述第一分类类别的第一生成卡通图像集，以使所述第一生成卡通图像集与所述第一样本图像集的集合作为第二样本图像集；

基于所述第二样本图像集训练所述第一初始模型，直至达到第二停止训练条件，得到图像转换模型。

2.根据权利要求1所述的方法，其特征在于，所述第二样本图像集的生成方式包括：

在属于所述第一分类类别的卡通图像的卡通风格，与属于所述第二分类类别的卡通图像的卡通风格不一致的情况下，在所述第一生成卡通图像集中确定所述第二数量的目标生成卡通图像，所述目标生成卡通图像的卡通风格与属于所述第二分类类别的卡通图像的卡通风格一致；

将所述目标生成卡通图像、所述第一样本图像集中的所述多个真人图像和属于所述第二分类类别的卡通图像的集合，作为所述第二样本图像集。

3.根据权利要求1所述的方法，其特征在于，所述第二样本图像集的生成方式包括：

在属于所述第一分类类别的卡通图像的卡通风格，与属于所述第二分类类别的卡通图像的卡通风格一致的情况下，计算所述第二数量与所述第一数量的差，得到第三数量；

在所述第一生成卡通图像集中确定所述第三数量的目标生成卡通图像，所述目标生成卡通图像的卡通风格与属于所述第二分类类别的卡通图像的卡通风格一致；

将所述目标生成卡通图像和所述第一样本图像集的集合作为所述第二样本图像集。

4.根据权利要求1所述的方法，其特征在于，所述第一数量与所述第二数量的比值为预设比值。

5.根据权利要求1所述的方法，其特征在于，所述基于所述第一样本图像集训练第一初始模型，直至达到第一停止训练条件，得到第二初始模型之后，还包括：

将属于第三分类类别的真人图像集输入至所述第二初始模型中，得到属于所述第三分类类别的第二生成卡通图像集，以使所述第一生成卡通图像集、所述第二生成卡通图像集与所述第一样本图像集的集合作为第二样本图像集。

6.根据权利要求1所述的方法，其特征在于，所述基于所述第二样本图像集训练所述第一初始模型，直至达到第二停止训练条件，得到图像转换模型之后，还包括：

获取待转换的目标真人图像；

通过所述图像转换模型，计算所述目标真人图像的注意力特征图，并基于所述注意力特征图设置归一化网络层的网络参数，基于完成网络参数设置的归一化网络层对所述注意力特征图进行解码，得到与所述目标真人图像对应的卡通图像。