TensorFlow学习记录：读取ckpt模型里面的张量名字和读取pb模型里面的张量名字

Barcelooooooooona

于 2019-08-02 10:57:14 发布

阅读量1.1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习 TensorFlow 文章标签： TensorFlow ckpt pb 模型读取

本文链接：https://blog.youkuaiyun.com/weixin_41137655/article/details/98179944

本文档记录了如何从TensorFlow的ckpt和pb模型中读取张量名称。针对ckpt模型，介绍了所需的文件组成，并提供了代码示例将所有张量名保存至txt文件。对于pb模型，同样给出了读取张量名称及查看其属性的代码，结果同样导出为txt文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一般情况下，我们得到一个模型后都想知道模型里面的张量，下面分别从ckpt模型和pb模型中读取里面的张量名字。

1.读取ckpt模型里面的张量

首先，ckpt模型需包含以下文件，一个都不能少

在这里插入图片描述
然后编写代码，将所有张量的名字都保存到tensor_name_list_ckpt.txt文件中

import tensorflow as tf

#直接读取图的结构，不需要手动重新定义 
meta_graph = tf.train.import_meta_graph("model.ckpt.meta")

with tf.Session()as sess:
	meta_graph.restore(sess,"D:/Face_recognition_github/20180402-114759/model.ckpt")

	tensor_name_list = [tensor.name for tensor in tf.get_default_graph().as_graph_def().node]
	with open("tensor_name_list_ckpt.txt"</