TensorFlow查看CKPT各节点名称实例
发布时间：2026-10-13 |
2人浏览过
本文介绍如何使用TensorFlow查看CKPT文件中各节点变量的名称。通过Python脚本读取模型文件，获取变量到形状映射，并打印出所有节点名称。
查看CKPT节点名称
运行以下脚本，可以打印出模型各个节点变量的名称：
from tensorflow.python import pywrap_tensorflow
import os
checkpoint_path=os.path.join('model.ckpt-131805')
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
checkpoint_path为模型路径
以上这篇TensorFlow查看CKPT各节点名称实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持本站。
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
