最近愁云惨淡,工作好难找。记录一下cycle GAN的代码笔记
代码参见:
leehomyc/cyclegan-1github.com
Cycle GAN的原理这里不多介绍,官方的网站也有许多有意思的demo,
https://junyanz.github.io/CycleGAN/junyanz.github.io
总体来说,cycle GAN的主要思想用下图足以表示:
有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的图