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之实现OCR识别
将图片翻译成文字一般被称为光学文字识别（Optical Character Recognition,OCR）,这篇文章主要给大家介绍了关于Python实现OCR识别的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考下
2021-08-08
深入flask之异步非堵塞实现代码示例
这篇文章主要介绍了深入flask之异步非堵塞实现代码示例,小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-07-07
Python文件操作和异常处理的方法和技巧
本文介绍了Python文件操作和异常处理的方法和技巧，包括读写文本文件、读写二进制文件、文件指针、异常类型和处理、try-except语句、异常处理程序等内容，帮助读者更好地掌握Python在文件操作和异常处理方面的应用
2023-05-05
详解Python使用Plotly绘图工具，绘制甘特图
这篇文章主要介绍了Python使用Plotly绘图工具，绘制甘特图，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-04-04
python知识:装饰器@property到底有啥用途
这篇文章主要介绍了python装饰器@property到底有啥用途，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2023-01-01
python GoogleIt库实现在Google搜索引擎上快速搜索
这篇文章主要为大家介绍了python GoogleIt库实现在Google搜索引擎上快速搜索功能探究,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2024-01-01
python实现图片九宫格分割
一张图片分成的九宫图，大家知道是怎么做到吗？这篇文章就为大家详细介绍了python实现图片九宫格分割功能，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2019-07-07
Python字符串格式化str.format()方法的实现
字符串的格式化是一个非常重要的功能，用于创建包含变量值的字符串，本来就来介绍一下Python字符串格式化str.format()方法的实现，感兴趣的可以了解一下
2023-11-11
Python爬虫定时计划任务的几种常见方法(推荐)
这篇文章主要介绍了Python爬虫定时计划任务的几种常见方法,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-01-01
Python中无限元素列表的实现方法
这篇文章主要介绍了Python中无限元素列表的实现方法,很实用的功能,需要的朋友可以参考下
2014-08-08