jerryli27 / TwinGAN

Twin-GAN -- Unpaired Cross-Domain Image Translation with Weight-Sharing GANs
Apache License 2.0
719 stars 98 forks source link

How to use inference/image_translation_infer.py in my trained model? #19

Open HiminDong opened 5 years ago

HiminDong commented 5 years ago

when i use my model model/128 to replace the model/256 ,

tf.flags.DEFINE_string("model_path", "model/128/", "Path containing a checkpoint.") tf.flags.DEFINE_integer('image_hw',128, 'height and width of the input image.') tf.flags.DEFINE_string('input_tensor_name','custom_input_ph', 'Optional input tensor name. e.g. sources_ph.') tf.flags.DEFINE_string('output_tensor_name','custom_generated_targets', 'e.g. custom_generated_t_style_source:0') tf.flags.mark_flags_as_required(['model_path', 'output_tensor_name']) FLAGS = tf.flags.FLAGS

I get an error: ValueError: Attempted to map inputs that were not found in graph_def: [custom_input_ph:0]