tensorflow打印模型图_Tensorflow 打印模型参数

可以再模型训练好后运行以下程序得到

variable_names = [v.name for v in tf.trainable_variables()]

values = sess.run(variable_names)

for k,v in zip(variable_names, values):

print("Variable: ", k)

print("Shape: ", v.shape)

print(v)

从ckpt文件加载得到

import tensoflow as tf

from tensorflow.python import pywrap_tensorflow

model_dir = "./ckpt/"

ckpt = tf.train.get_checkpoint_state(model_dir)

ckpt_path = ckpt.model_checkpoint_path

reader = pywrap_tensorflow.NewCheckpointReader(ckpt_path)

param_dict = reader.get_variable_to_shape_map()

for key, val in param_dict.items():

try:

print key, val

except:

pass

从pb文件中打印参数

with tf.Graph().as_default():

graph0 = tf.GraphDef()

with open( pbName, mode='rb') as f:

grap