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

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('path/to/event/file', 'tag0')

请注意，这里假定事件文件是完全写入的 - 如果不是，则可能需要一些错误处理。