pb文件使用tensorboard查看网络结构

Posted 2023-03-27

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pb文件使用tensorboard查看网络结构相关的知识，希望对你有一定的参考价值。

参考技术A 新建generate_graph.py文件

运行python generate_graph.py，此时会在log目录下生成一个log文件。

此时会出现
TensorBoard 1.13.1 at http://127.0.0.1:6006 (Press CTRL+C to quit)
打开浏览器，输入http://127.0.0.1:6006，回车，可以看到网络结构。

使用TensorBoard

偶然机会需要使用ckpt转换为pb，所以需要查找到输出节点。

指定ckpt文件，然后会在当前文件夹下生成__tb文件夹，里面还有一个文件。

import tensorflow as tf


sess = tf.Session()
tf.train.import_meta_graph("F:/ModelArts/cityscapesScripts/exp/train_on_train_set/train/model.ckpt-1000.meta")
tf.summary.FileWriter("__tb", sess.graph)

切换到指定目录

技术图片