TensorFlow 2.0 Beta is available Learn more

tf.contrib.gan.CycleGANModel

View source on GitHub

Class CycleGANModel

An CycleGANModel contains all the pieces needed for CycleGAN training.

The model model_x2y generator F maps data set X to Y, while the model model_y2x generator G maps data set Y to X.

See https://arxiv.org/abs/1703.10593 for more details.

Args:

  • model_x2y: A GANModel namedtuple whose generator maps data set X to Y.
  • model_y2x: A GANModel namedtuple whose generator maps data set Y to X.
  • reconstructed_x: A Tensor of reconstructed data X which is G(F(X)).
  • reconstructed_y: A Tensor of reconstructed data Y which is F(G(Y)).

Properties

model_x2y

model_y2x

reconstructed_x

reconstructed_y