transfer.py 文件源码

def transfer_summary(vgg, loss, content_shape):
    """
    summaries of loss and result image.
    """
    image = tf.add(vgg.upstream, VggNet.mean_color_bgr())
    image = tf.image.resize_images(image, content_shape)
    image = tf.saturate_cast(image, tf.uint8)
    image = tf.reverse(image, [3])

    summary_image = tf.summary.image('generated image', image, max_outputs=1)

    summary_loss = tf.summary.scalar('transfer loss', loss)

    return tf.summary.merge([summary_image, summary_loss])