Tensorflow FileWriter 不写入（未找到图像数据）

Question

我正在尝试编写将显示我的 matplotlib 图形的代码。当我运行下面的代码生成一个事件文件时，但是当我运行 tensorboard 并将 logdir 指向我得到的文件夹时

No image data was found.

示例代码：

import numpy as np
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt

import tensorflow as tf
import io

class bla():
    def __init__(self, sess):
        self.sess = sess

class Dummy():
    def __init__(self, sess):
        self.fill_scheme = {
            'adv': np.linspace(0, 20),
            'butt': np.linspace(0, 20) * -1
        }
        self.model = bla(sess)

    def test(self, name=100):
        checker = hasattr(self.model, 'sess')
        writer = tf.summary.FileWriter('./im_data', graph=None if not checker else self.model.sess.graph)
        summary = tf.Summary()
        for k, v in self.fill_scheme.items():
            plt.figure()
            plt.plot(v)
            s = io.BytesIO()
            plt.savefig(s)
            plt.clf()
            img_summary = tf.Summary.Image(encoded_image_string=s.getvalue())
            summary.value.add(tag='im/it_{}_{}'.format(name, k), image=img_summary)
            writer.add_summary(summary, name)
            plt.close()
        writer.close()

sess = tf.InteractiveSession()
a = Dummy(sess)
a.test()

这不是我的错误，所以我不确定出了什么问题？

Answer 1

您使用的是哪个版本的 TF 和 Tensorboard？当我运行你的代码似乎一切正常时（TF 1.8，Tensorboard 1.7.0）。

与 "tensorboard --logdir im_data/" 我得到以下信息：

Tensorflow FileWriter 不写入（未找到图像数据）

Tensorflow FileWriter not writing(No image data was found)

tensorflow

tensorboard