tensorboard

最新推荐文章于 2024-07-22 22:02:52 发布

joyicejin

最新推荐文章于 2024-07-22 22:02:52 发布

阅读量307

点赞数

CC 4.0 BY-SA版权

分类专栏： tensorboard

本文链接：https://blog.youkuaiyun.com/qq_32450111/article/details/81317070

tensorboard 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用TensorBoard将导入的Protobuf（.pb）模型可视化为图的方法。通过定义函数import_to_tensorboard，可以指定模型位置和日志目录，实现模型的可视化展示。在指定的会话中，读取并解析模型文件，然后将其添加到TensorBoard的日志中，最后通过启动TensorBoard，可以查看模型的结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这里写图片描述

https://github.com/tensorflow/tensorboard/blob/master/README.md

file_writer = tf.summary.FileWriter(‘/path/to/logs’, sess.graph)

tensorboard –logdir path/to/logs

`"""Imports a protobuf model as a graph in Tensorboard."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function


from tensorflow.core.framework import graph_pb2
from tensorflow.python.client import session
from tensorflow.python.framework import importer
from tensorflow.python.framework import ops
from tensorflow.python.platform import app
from tensorflow.python.platform import gfile
from tensorflow.python.summary import summary


def import_to_tensorboard(model_dir, log_dir):
  """View an imported protobuf model (`.pb` file) as a graph in Tensorboard.

  Args:
    model_dir: The location of the protobuf (`pb`) model to visualize
    log_dir: The location for the Tensorboard log to begin visualization from.

  Usage:
    Call this function with your model location and desired log directory.
    Launch Tensorboard by pointing it to the log directory.
    View your imported `.pb` model as a graph.
  """
  with session.Session(graph=ops.Graph()) as sess:
    with gfile.FastGFile(model_dir, "rb") as f:
      graph_def = graph_pb2.GraphDef()
      # graph_def.ParseFromString(f.read())
      importer.import_graph_def(graph_def)

    pb_visual_writer = summary.FileWriter(log_dir)
    pb_visual_writer.add_graph(sess.graph)
    print("Model Imported. Visualize by running: "
          "tensorboard --logdir={}".format(log_dir))


if __name__ == "__main__":
  model_dir="/media/joyice/ssdata/cap-git-v1/model/cap49.pbtxt"
  log_dir="/media/joyice/ssdata/cap-git-v1/log"
  import_to_tensorboard(model_dir, log_dir)