train.py 文件源码

def build_image_summary(self):
        """
        A simple graph for write image summary
        :return:
        """
        log_image_data = tf.placeholder(tf.uint8, [None, None, 3])
        log_image_name = tf.placeholder(tf.string)
        log_image = tf.image_summary(log_image_name, tf.expand_dims(log_image_data, 0), max_images=1)
        # log_image = tf.image_summary(log_image_name, log_image_data, max_images=50)
        return log_image, log_image_data, log_image_name