cyclegan网络结构_Cycle GAN代码笔记

  • Post author:
  • Post category:其他


最近愁云惨淡,工作好难找。记录一下cycle GAN的代码笔记

代码参见:

leehomyc/cyclegan-1​github.com

f714ba36305eeb81b618ea0c3747e75c.png

Cycle GAN的原理这里不多介绍,官方的网站也有许多有意思的demo,

https://junyanz.github.io/CycleGAN/​junyanz.github.io

总体来说,cycle GAN的主要思想用下图足以表示:

86cbf711e0c2796c39c4f7f9b35d7b8f.png

有AB两个域的图片,即如上图的马和斑马。

对于A域:

domain A的图片经过鉴别器A得到 prob_real_a_is_real

domain A的图片经过生成器A2B得到 fake_images_b

图片 fake_images_b经过鉴别器B得到 prob_fake_b_is_real

图片 fake_images_b 经过生成器B2A得到 cycle_images_a

对于B域:

domain B的图片经过鉴别器B得到 prob_real_b_is_real

domain B的图