模型训练方法、图像风格转换方法、装置及电子设备

专利主权项内容

1.一种风格转换模型的训练方法，其特征在于，包括：确定第一样本集，所述第一样本集包括多个样本内容图像和多个样本风格图像；根据第一样本集对第一初始模型进行对抗训练，得到训练后的第一初始模型，所述第一初始模型包括第一生成器，所述第一生成器用于对输入的原始图像和参考图像均进行全局特征和局部特征增强，并将特征增强后的原始图像和参考图像进行融合，获得转化图像；根据所述训练后的第一初始模型，获得风格转换模型，所述风格转换模型包括所述训练后的第一初始模型中的第一生成器；其中，当原始图像为样本内容图像时，参考图像为样本风格图像，转化图像为风格化内容图像，所述风格化内容图像是指将样本内容图像的风格，迁移为样本风格图像的风格后的图像。