Commit 34da9a1f authored by Yuxin Wu's avatar Yuxin Wu

bug fix in im2im

parent cb5e7e16
...@@ -118,7 +118,7 @@ class Model(ModelDesc): ...@@ -118,7 +118,7 @@ class Model(ModelDesc):
fake_output = tf.image.grayscale_to_rgb(fake_output) fake_output = tf.image.grayscale_to_rgb(fake_output)
viz = (tf.concat(2, [input, output, fake_output]) + 1.0) * 128.0 viz = (tf.concat(2, [input, output, fake_output]) + 1.0) * 128.0
viz = tf.cast(tf.clip_by_value(viz, 0, 255), tf.uint8, name='viz') viz = tf.cast(tf.clip_by_value(viz, 0, 255), tf.uint8, name='viz')
tf.image_summary('gen', viz, max_outputs=max(30, BATCH)) tf.summary.image('gen', viz, max_outputs=max(30, BATCH))
all_vars = tf.trainable_variables() all_vars = tf.trainable_variables()
self.g_vars = [v for v in all_vars if v.name.startswith('gen/')] self.g_vars = [v for v in all_vars if v.name.startswith('gen/')]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment