使用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)

切换到指定目录

 

 最后会有一个链接

 

 在浏览器中复制打开