- summaries.append(tf.scalar_summary('total_loss', total_loss))
- summaries.append(tf.scalar_summary('reproduction_loss', reproduction_loss))
- # summaries.append(tf.scalar_summary('prior_loss', prior_loss))
- summaries.append(tf.image_summary('Input Image', input_placeholder, 3))
- summaries.append(tf.image_summary('Output Image', prediction, 3))
- summaries.append(tf.image_summary('Target Image', target_placeholder, 3))
+ summaries.append(tf.summary.scalar('total_loss', total_loss))
+ summaries.append(tf.summary.scalar('reproduction_loss', reproduction_loss))
+ # summaries.append(tf.summary.scalar('prior_loss', prior_loss))
+ summaries.append(tf.summary.image('Input Image (before)', input_placeholder[:, :, :, 0:3], 3));
+ summaries.append(tf.summary.image('Input Image (after)', input_placeholder[:, :, :, 3:6], 3));
+ summaries.append(tf.summary.image('Output Image', prediction, 3))
+ summaries.append(tf.summary.image('Target Image', target_placeholder, 3))