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

切换到指定目录

使用TensorBoard

 

 最后会有一个链接

使用TensorBoard

 

 在浏览器中复制打开

使用TensorBoard

 

上一篇:ckpt模型与.pb文件互相转换


下一篇:cs20_6-1