一种图像风格迁移模型训练方法以及图像风格迁移方法

摘要

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