使用Tensorboard展示Tensorflow模型的图结构
  Os1Mb0Rs86a5 2023年11月02日 33 0

1 将ckpt的meta文件转换成tensorboard需要的log文件

参考代码如下:

import tensorflow as tf


def write_ckpt_meta_to_tensorboard(ckpt_meta_file, log_dir):
    g = tf.Graph()

    with g.as_default() as g:
        tf.train.import_meta_graph(ckpt_meta_file)

    with tf.Session(graph=g) as sess:
        file_writer = tf.summary.FileWriter(logdir=log_dir, graph=g)
        file_writer.close()


if __name__ == "__main__":
    meta_file = "/tmp/model.ckpt-806.meta"
    log_dir = "/tmp/tensorboard_log_806"
    write_ckpt_meta_to_tensorboard(meta_file, log_dir)

运行上面的代码,会产生log文件tensorboard_log_806.

2.启动Tensorboard服务

接着运行下面的命令启动tensorboard服务,前提是安装了tensorboard库.运行下面的pip命令进行安装

# 安装tensorboard
pip install tensorboard


# 启动tensorboard服务
tensorboard --logdir=/tmp/tensorboard_log_806

3.使用浏览器访问Tensorboard的Web

打开浏览器输入: localhost:6006, 即可查看图模型

【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读
Os1Mb0Rs86a5