[发明专利]一种图像风格迁移模型训练方法以及图像风格迁移方法

摘要

本申请提供了一种图像风格迁移模型训练方法以及图像风格迁移方法，图像风格迁移模型训练方法包括：获取风格参考图像以及内容图像；将风格参考图像以及内容图像输入第一神经网络，为内容图像提取第一特征向量，并为风格参考图像提取第二特征向量；基于第一特征向量对内容图像进行还原，得到内容图像的迁移图像；将风格参考图像和迁移图像输入第一神经网络，为迁移图像提取第三特征向量；基于第二特征向量和第三特征向量，计算风格参考图像以及迁移图像之间的色调损失；根据色调损失，对第一神经网络进行训练。该方法能够以更快的速度得到训练好的图像风格迁移模型。

主权项

1.一种图像风格迁移模型训练方法，其特征在于，包括：获取风格参考图像以及内容图像；将所述风格参考图像以及所述内容图像输入第一神经网络，为所述内容图像提取第一特征向量，并为所述风格参考图像提取第二特征向量；基于所述第一特征向量对所述内容图像进行还原，得到所述内容图像的迁移图像；将所述风格参考图像和所述迁移图像输入所述第一神经网络，为所述迁移图像提取第三特征向量；基于所述第二特征向量和所述第三特征向量，计算所述风格参考图像以及所述迁移图像之间的色调损失；根据所述色调损失，对所述第一神经网络进行训练。