Cycle GAN

CycleGan class

docclean.cycle_gan.CycleGan(checkpoint_path=None, restore_checkpoint=True)

Cycle GAN in tensorflow based on the pix2pix.

Args:

checkpoint_path: Where to put the checkpoints

restore_checkpoint: Restore old checkpoints or not

---

train method

CycleGan.train(dirty_images, clean_images, epochs=50)

Training function.

Args:

dirty_images (tf.data.Dataset): dirty images dataset

clean_images (tf.data.Dataset): clean images dataset

epochs (int): Number of epochs to run

---