查看TensorFlow中checkpoint内变量的几种方法
checkpoint 文件有两个版本,关于两个版本的不同,请移步:https://blog.csdn.net/u014061630/article/details/80698843
查看ckpt中变量的方法有三种:
在有model的情况下,使用tf.train.Saver进行restore
使用tf.train.NewCheckpointReader直接读取ckpt文件,这种方法不需要model。
使用tools里的freeze_graph来读取ckpt
注意:
- 如果模型保存为.ckpt的文件,则使用该文件就可以查看.ckpt文件里的变量。ckpt路径为 model.ckpt
- 如果模型保存为.ckpt-xxx-data (图结构)、.ckpt-xxx.index (参数名)、.ckpt-xxx-meta (参数值)文件,则需要同时拥有这三个文件才行。并且ckpt的路径为 model.ckpt-xxx
1. 基于model来读取ckpt文件里的变量
1.首先建立model
2.从ckpt中恢复变量
with tf.Graph().as_default() as g:
#建立mode