python - 如何从张量板事件摘要中提取和保存图像？

Question

给定一个 tensorflow 事件文件，我如何提取与特定标签对应的图像，然后以通用格式将它们保存到磁盘，例如.png?

Answer 1

您可以像这样提取图像。输出格式可能取决于摘要中图像的编码方式，因此写入磁盘的结果可能需要使用 .png

之外的其他格式

import os
import scipy.misc
import tensorflow as tf

def save_images_from_event(fn, tag, output_dir='./'):
    assert(os.path.isdir(output_dir))

    image_str = tf.placeholder(tf.string)
    im_tf = tf.image.decode_image(image_str)

    sess = tf.InteractiveSession()
    with sess.as_default():
        count = 0
        for e in tf.train.summary_iterator(fn):
            for v in e.summary.value:
                if v.tag == tag:
                    im = im_tf.eval({image_str: v.image.encoded_image_string})
                    output_fn = os.path.realpath('{}/image_{:05d}.png'.format(output_dir, count))
                    print("Saving '{}'".format(output_fn))
                    scipy.misc.imsave(output_fn, im)
                    count += 1  

然后一个示例调用可能如下所示:

save_images_from_event('路径/到/事件/文件', 'tag0')

请注意，这假设事件文件已完全写入——如果没有，则可能需要进行一些错误处理。