StarGAN中的损失函数

学习笔记
论文地址:http://openaccess.thecvf.com/content_cvpr_2018/papers/Choi_StarGAN_Unified_Generative_CVPR_2018_paper.pdf

我们用生成器G把输入的图像x转换为目标域c,通过生成器G输出图像y,即G(x,c)→y。为了便生成器G可以学会灵活地转换输入图像,我们随机生成的目标域标签c。我们的判别器在源和域标签上产生概率分布为在这里插入图片描述

损失函数

一、对抗损失:
在这里插入图片描述
使用GAN的常规函数,生成器G通过输入图像x和目标域标签c生成图像G(x,c),而判别器D来辨别真实图像与生成图像。生成器G试图最小化该目标,而鉴别器D试图使其最大化。
实际操作中换成了WGAN的对抗损失:
在这里插入图片描述

二、类别损失:

判别器
在这里插入图片描述
我们的目地是将输入图像x转换为输出图像y,让正确地分类到目标域c。
Dcls(c’| x)代表判别器将真实样本归为原始标签c’ 的概率分布,判别器D的目标是最小化Loss 。图像x和和原域标签c’是由训练集给出的。

生成器
在这里插入图片描述
使生成器G生成假图片(x’)让它尽可能被判别器D分类成目标域c(比如愤怒),因此最小化Loss

重建损失
最小化损失并不能保证翻译的图像仅改变输入图片的与域相关的信息部分,而不改变其输入图像的内容。因此我们在此加上一个重建损失。
在这里插入图片描述
将G(x,c)和图片x的原始标签c’结合喂入到G中,将生成的图片和x计算1范数差异。(1范数就是向量中非零元素的绝对值之和,L1范数可以度量两个向量间的差异)

  • 3
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值