This tutorial trains a model to translate from images of horses, to images of zebras.

pip install git+ import tensorflow as tfįrom tensorflow_2pix import pix2pix Install the tensorflow_examples package that enables importing of the generator and the discriminator. All you need is the source and the target dataset (which is simply a directory of images). This opens up the possibility to do a lot of interesting tasks like photo-enhancement, image colorization, style transfer, etc. In other words, it can translate from one domain to another without a one-to-one mapping between the source and target domain. The code for CycleGAN is similar, the main difference is an additional loss function, and the use of unpaired training data.ĬycleGAN uses a cycle consistency loss to enable training without the need for paired data. This notebook assumes you are familiar with Pix2Pix, which you can learn about in the Pix2Pix tutorial. The paper proposes a method that can capture the characteristics of one image domain and figure out how these characteristics could be translated into another image domain, all in the absence of any paired training examples.

This notebook demonstrates unpaired image to image translation using conditional GAN's, as described in Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks, also known as CycleGAN.

Zebra 2 tutorial