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

基本信息

申请号 CN201810404755.6 申请日 -
公开(公告)号 CN108596830B 公开(公告)日 2022-04-22
申请公布号 CN108596830B 申请公布日 2022-04-22
分类号 G06T3/00(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 孙源良;刘萌;樊雨茂;李彩虹 申请(专利权)人 国信优易数据股份有限公司
代理机构 北京超凡志成知识产权代理事务所(普通合伙) 代理人 徐丽
地址 100000 北京市丰台区南四环西路188号一区31号楼1层101-8(园区)
法律状态 -

摘要

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