pix2pix.py 文件源码

def build_summaries(model):
    """
    build image summary: [source batch, target batch, result batch]
    """
    keys = ['source_images', 'target_images', 'output_images']

    images = tf.concat([model[k] for k in keys], axis=2)

    images = tf.reshape(images, [1, FLAGS.batch_size * 256, 768, 3])

    images = tf.saturate_cast(images * 127.5 + 127.5, tf.uint8)

    summary = tf.summary.image('images', images, max_outputs=4)

    return {
        'summary': summary,
    }