pix2pix.py 文件源码

def build_output(model):
    """
    save translation result to FLAGS.target_image_path.
    """
    images = tf.concat(
        [model['source_images'], model['output_images']], axis=2)

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

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

    images = tf.image.encode_png(images)

    return tf.write_file(FLAGS.target_image_path, images)