其他分享
首页 > 其他分享> > 使用TensorBoard

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

切换到指定目录

 

 最后会有一个链接

 

 在浏览器中复制打开

 

标签:__,sess,ckpt,train,meta,TensorBoard,使用,tf
来源: https://www.cnblogs.com/birdmmxx/p/12913223.html