跟李沐学AI:样式迁移

样式迁移需要两张输入图像:一张是内容图像,另一张是样式图像。 我们将使用神经网络修改内容图像,使其在样式上接近样式图像,得到合成图片。类似手机相册中的滤镜效果。

奠基性工作:基于CNN的样式迁移 

任务:训练一个合成图片,融合另外两张图片的样式和内容。

首先,我们初始化合成图像,例如将其初始化为内容图像。合成图像是风格迁移过程中唯一需要更新的变量,即样式迁移所需迭代的模型参数。

然后,我们选择一个预训练的卷积神经网络来抽取图像的特征,其中的模型参数在训练中无须更新,我们可以选择其中某些层的输出作为内容特征或风格特征。如图中所示,以一个含有三个卷积层的与训练神经网络抽取内容图像和样式图像的特征,其中预训练模型的第二层输出为内容特征,第一层与第三层的输入为风格特征。

随后,通过前向传播计算样式迁移的两类损失函数(样式、风格),通过反向传播迭代模型参数,即不断更新合成图像。样式迁移常用的损失函数又三部分组成:1. 内容损失使合成图像与内容图像再内容特征上相近;2. 样式损失使合成图像与样式图像在样式特征中相近;3. 总变差损失以减少合成图像中的噪点

  • 9
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值