无法使用TensorFlow 0.7.1在TensorBoard中显示Inception v3模型

时间:2016-02-25 18:17:55

标签: tensorflow tensorboard

我试图在TensorFlow 0.7.1中使用TensorBoard可视化Google的Inception v3模型,但我无法这样做。 TensorBoard Graph选项卡与语句

一起停止
  

数据:阅读graph.pbtxt

docs

我下载了{Targe Data : Reading graph.pbtxt。图表protobuffer位于import os import os.path import tensorflow as tf from tensorflow.python.platform import gfile INCEPTION_LOG_DIR = '/tmp/inception_v3_log' if not os.path.exists(INCEPTION_LOG_DIR): os.makedirs(INCEPTION_LOG_DIR) with tf.Session() as sess: model_filename = '/tmp/imagenet/classify_image_graph_def.pb' with gfile.FastGFile(model_filename, 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) _ = tf.import_graph_def(graph_def, name='') writer = tf.train.SummaryWriter(INCEPTION_LOG_DIR, graph_def) writer.close()

这是我转储模型的代码:

events.out.tfevents.1456423256.[hostname]

这会转储一个名为tensorboard --logdir /tmp/inception_v3_log 的91 MB文件(与图形protobuffer大小相同),因此图表似乎在某处。

我按如下方式运行TensorBoard:

__attribute_warn_unused_result__

这导致图表页面上的上述挂载栏。

Chrome JavaScript控制台会出现此错误:

  

未捕获的TypeError:无法读取未定义的属性“0”

我认为这与图表缺失的事实有关。

我在OS X 10.11.3上尝试使用Chrome 48.0.2564.116(64位),使用TensorFlow 0.7.1,使用Bazel构建Python 3,使用TensorFlow 0.7.1,通过pip安装Python 2相同的结果。

我还验证了我可以可视化使用inception v3 model生成的图形,因此这是一个专门针对Inception模型的问题。

2 个答案:

答案 0 :(得分:0)

丹尼尔,

我不知道你提到的protobuffer的东西,但我想你可能想要卸载protobuf并重新安装张量流。

升级(在Ubuntu上)从tensorflow v0.6升级到v0.7.1后,当我在启动张量板时遇到问题,我发现了以下帖子。

我认为protobuf存在问题。在这里谈到:https://github.com/tensorflow/tensorflow/issues/1134#issuecomment-185279000

我最终卸载了protobuf并重新安装了tensorboard。现在我可以启动电路板并查看我的图表。祝好运! :)

答案 1 :(得分:0)

使用以下代码: 这对我有用:

import tensorflow as tf
from tensorflow.python.platform import gfile
with tf.Session() as sess:
    model_filename ='YouGraphNameWithPath.pb'
    with gfile.FastGFile(model_filename, 'rb') as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        g_in = tf.import_graph_def(graph_def)
LOGDIR='PathWhereSummaryWillBeSaved'
train_writer = tf.summary.FileWriter(LOGDIR)
train_writer.add_graph(sess.graph)