[发明专利]一种图像风格迁移模型训练方法、装置及电子设备

说明书

本申请实施例提供的一种图像风格迁移模型训练方法、装置及电子设备，通过将样本源域图像和样本目标域图像输入待训练的图像风格迁移模型；通过内容编码层对样本源域图像和样本目标域图像进行特征提取，得到源内容张量和源风格张量；通过风格编码层对样本源域图像和样本目标域图像进行特征提取，得到目标内容张量和目标风格张量；通过预设损失函数，根据源内容张量、源风格张量、目标内容张量和目标风格张量计算待训练的图像风格迁移模型的当前损失；根据当前损失对待训练的图像风格迁移模型的参数进行调整，直至当前损失小于预设阈值，得到训练好的图像风格迁移模型。实现了提取样本图像的多个维度的特征进行模型的训练，避免模型的内容敏感性。

技术领域

本申请涉及模型训练的技术领域，特别是涉及一种图像风格迁移模型训练方法、装置及电子设备。

背景技术

目前，图像风格迁移模型在电影业、摄影技术、时尚、美容相机和电子商务等领域，已经有着广泛的应用，通过图像风格迁移模型可以自动的将一幅图像转化为另一种风格的图像。

然而，当前图像风格迁移模型在训练过程中，往往只提取样本图像单一维度的特征，然后根据提取的特征进行模型的训练，从而导致训练得到的模型对图像的内容较为敏感，所需要的样本数据的数据量也较多。

发明内容

本申请实施例的目的在于提供一种图像风格迁移模型训练方法、装置及电子设备，用以解决现有技术中训练图像风格迁移模型的过程中，样本数据需求多的问题。具体技术方案如下：

本申请实施例的第一方面，首先提供了一种图像风格迁移模型训练方法，上述方法包括：

将样本源域图像和样本目标域图像输入待训练的图像风格迁移模型，其中，图像风格迁移模型包括内容编码层和风格编码层；

通过内容编码层对样本源域图像和样本目标域图像进行特征提取，得到源内容张量和目标域内容张量；

通过风格编码层对样本源域图像和样本目标域图像进行特征提取，得到源风格张量和目标风格张量；

通过预设损失函数，根据源内容张量、源风格张量、目标内容张量和目标风格张量计算待训练的图像风格迁移模型的当前损失；

根据当前损失对待训练的图像风格迁移模型的参数进行调整，返回将样本源域图像和样本目标域图像输入待训练的图像风格迁移模型的步骤继续执行，直至当前损失小于预设阈值，得到训练好的图像风格迁移模型。

可选的，通过预设损失函数，根据源内容张量、源风格张量、目标内容张量和目标风格张量计算待训练的图像风格迁移模型的当前损失，包括：

通过预设损失函数，根据源内容张量、源风格张量、目标内容张量和目标风格张量计算待训练的图像风格迁移模型的项目损失，其中，项目损失包括重构损失、潜在语义损失、循环重构损失、对抗损失、风格多样性损失和感知损失中的至少一种；

根据项目损失计算得到当前损失。

可选的，重构损失的计算方式包括：通过预设生成器，根据源内容张量和源风格张量计算得到源域域内重构图；通过预设生成器，根据目标内容张量和目标风格张量计算得到目标域域内重构图；根据源域域内重构图和目标域域内重构图，通过第一预设损失函数计算得到重构损失。