【时间】2019.12.03
【题目】tensorflow的模型保存文件.ckpt包含什么以及如何读取、并将其转换为Keras的.h5文件
参考:tensorflow中读取模型中保存的值, tf.train.NewCheckpointReader
一、tensorflow的模型保存文件.ckpt
.ckpt即checkpoint文件。
一个标准的.ckpt模型文件以下文件:
model.ckpt文件组成:
model.ckpt.meta
model.ckpt.data-00000-of-00001
model.ckpt.index
二、读取.ckpt文件中的内容
使用 NewCheck =tf.train.NewCheckpointReader(ckpt_file)读取:
import tensorflow as tf
import pprint # 使用pprint 提高打印的可读性
NewCheck =tf.train.NewCheckpointReader("model.ckpt")
2.1.打印模型中的所有变量
print("debug_string:\n")
pprint.pprint(NewCheck.debug_string().decode("utf-8")) #类型是str
其中有3个字段, 分别是名字name, 数据类型type,