查看ckpt的各个节点
from tensorflow.python import pywrap_tensorflow
# checkpoint_path = 'model.ckpt-xxx'
checkpoint_path = './model'
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print("tensor_name: ", key)
print(list(var_to_shape_map.keys()))
node_names=list(var_to_shape_map.keys())
转换ckpt为pb:
import tensorflow as tf
from tensorflow import graph_util
def freeze_graph(ckpt, output_graph):
# saver = tf.train.Saver()
saver = tf.train.import_meta_graph(ckpt+'.meta', clear_devices=True)
with tf.Session() as sess:
# saver = tf.compat.v1.train.import_meta_graph(ckpt+'.meta', clear_devices=True)
saver.restore(sess, ckpt+'')