tf.contrib.gan.cyclegan_model( generator_fn, discriminator_fn, data_x, data_y, generator_scope='Generator', discriminator_scope='Discriminator', model_x2y_scope='ModelX2Y', model_y2x_scope='ModelY2X', check_shapes=True )
Defined in tensorflow/contrib/gan/python/train.py
.
Returns a CycleGAN model outputs and variables.
See https://arxiv.org/abs/1703.10593 for more details.
generator_fn
: A python lambda that takes data_x
or data_y
as inputs and returns the outputs of the GAN generator.discriminator_fn
: A python lambda that takes real_data
/generated data
and generator_inputs
. Outputs a Tensor in the range [-inf, inf].data_x
: A Tensor
of dataset X. Must be the same shape as data_y
.data_y
: A Tensor
of dataset Y. Must be the same shape as data_x
.generator_scope
: Optional generator variable scope. Useful if you want to reuse a subgraph that has already been created. Defaults to 'Generator'.discriminator_scope
: Optional discriminator variable scope. Useful if you want to reuse a subgraph that has already been created. Defaults to 'Discriminator'.model_x2y_scope
: Optional variable scope for model x2y variables. Defaults to 'ModelX2Y'.model_y2x_scope
: Optional variable scope for model y2x variables. Defaults to 'ModelY2X'.check_shapes
: If True
, check that generator produces Tensors that are the same shape as data_x
(data_y
). Otherwise, skip this check.A CycleGANModel
namedtuple.
ValueError
: If check_shapes
is True and data_x
or the generator output does not have the same shape as data_y
.
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/gan/cyclegan_model