Tensorflow: 从checkpoint文件中读取tensor方式
更新时间：2020年02月10日 14:49:57 作者：noirblack
今天小编就为大家分享一篇Tensorflow: 从checkpoint文件中读取tensor方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
在使用pre-train model时候，我们需要restore variables from checkpoint files.
经常出现在checkpoint 中找不到”Tensor name not found”.
这时候需要查看一下ckpt中到底有哪些变量
import os
from tensorflow.python import pywrap_tensorflow
checkpoint_path = os.path.join(model_dir, "model.ckpt")
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
print("tensor_name: ", key)
print(reader.get_tensor(key))
可以显示ckpt中的tensor名字和值，当然也可以用pycharm调试。
以上这篇Tensorflow: 从checkpoint文件中读取tensor方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
在用Python进行数据处理的时候，经常会遇到DataFrame中的某一列本应该是数值类型，但由于数据不规范导致在字段中夹杂了非数值类型，本文就介绍了Python如何判断字符串是否仅包含数字，感兴趣的可以了解一下2022-03-03
去马赛克是图像处理中的一项技术,用于从单色彩滤光片阵列（CFA）图像恢复全彩图像,本文介绍了一种基于简单线性插值的去马赛克算法,并展示了如何将MATLAB代码转换为Python代码,需要的朋友可以参考下2024-10-10
在本文中小编给大家整理了关于python全栈要学什么以及python全栈学习路线的知识点内容，需要的朋友们参考下。2019-06-06
这篇文章主要介绍了matlab中imadjust函数的作用及应用举例，本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下2020-02-02
这篇文章主要介绍了python使用内存zipfile对象在内存中打包文件示例,需要的朋友可以参考下2014-04-04
今天小编就为大家分享一篇Python 文本文件内容批量抽取实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
django框架自定义模板标签(template tag)操作示例
这篇文章主要介绍了django框架自定义模板标签(template tag)操作,结合实例形式分析了Django框架自定义模板标签原理、操作步骤与相关实现技巧,需要的朋友可以参考下2019-06-06
这篇文章主要给大家介绍了关于python更换国内镜像源三种实用方法的相关资料,更换Python镜像源可以帮助解决使用pip安装包时速度过慢或无法连接的问题,需要的朋友可以参考下2023-09-09
由于工作需要,客户需要将多个excel文件合并成一个excel中,下面这篇文章主要给大家介绍了关于使用Python快速进行Excel合并的几种场景,文中通过代码介绍的非常详细,需要的朋友可以参考下2023-10-10
Python语法定义了用于在 Python 编程中创建句子的所有规则集。如果想更深入地研究 Python 词法结构，需要了解构成语句的句法元素，即构成 Python 程序的基本单元，涵盖控制结构，在不同代码组之间引导程序流的构造，快跟随小编一起学习一下吧2022-04-04
最新评论