偶然机会需要使用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)
切换到指定目录
最后会有一个链接
在浏览器中复制打开