tensorflow实现读取模型中保存的值 tf.train.NewCheckpointReader
更新时间：2020年02月10日 11:10:06 作者：onexming
今天小编就为大家分享一篇tensorflow实现读取模型中保存的值 tf.train.NewCheckpointReader，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
使用tf.trian.NewCheckpointReader(model_dir)
一个标准的模型文件有一下文件, model_dir就是MyModel(没有后缀)
checkpoint Model.meta Model.data-00000-of-00001 Model.index
import tensorflow as tf
import pprint # 使用pprint 提高打印的可读性
NewCheck =tf.train.NewCheckpointReader("model")
打印模型中的所有变量
print("debug_string:\n")
pprint.pprint(NewCheck.debug_string().decode("utf-8"))
其中有3个字段, 分别是名字, 数据类型, shape
获取变量中的值
print("get_tensor:\n")
pprint.pprint(NewCheck.get_tensor("D/conv2d/bias"))
print("get_variable_to_dtype_map\n")
pprint.pprint(NewCheck.get_variable_to_dtype_map())
print("get_variable_to_shape_map\n")
pprint.pprint(NewCheck.get_variable_to_shape_map())
以上这篇tensorflow实现读取模型中保存的值 tf.train.NewCheckpointReader就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
在开发过程中，配置文件是少不了的，而且配置文件是有专门的格式的，比如：ini, yaml, toml 等等。而对于 Python 而言，也都有相应的库来解析相应格式的文件，下面我们来看看 ini 文件要如何解析2022-09-09
每次在读或写文件时,获取脚本文件的绝对路径和相对路径时很容易搞混,下面这篇文章主要给大家介绍了关于python文件读取和导包的绝对路径、相对路径的相关资料,需要的朋友可以参考下2022-04-04
Python 中eval()函数的正确使用及其风险分析(使用示例)
eval()是一个功能强大的工具,但使用时必须非常小心,了解其工作原理和潜在的风险是确保安全使用的关键,通过遵循上述建议,可以在享受eval()带来的便利的同时,最大限度地减少安全风险,本文介绍Python 中`eval()`函数的正确使用及其风险分析,感兴趣的朋友一起看看吧2024-07-07
这篇文章主要介绍了Python嵌套函数，作用域与偏函数用法,结合实例形式分析了Python嵌套函数，作用域与偏函数的功能、定义与相关使用方法,需要的朋友可以参考下2019-12-12
Tensorflow的可视化工具Tensorboard的初步使用详解
这篇文章主要介绍了Tensorflow的可视化工具Tensorboard的初步使用详解，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-02-02
这篇文章主要为大家详细介绍了如何利用Python编程语言和wxPython模块,打开指定文件夹中的DOC文档,并将它们的内容合并成一个便捷的TXT文档,需要的可以参考下2024-03-03
这篇文章主要介绍了python 带时区的日期格式化操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-10-10
django获取from表单multiple-select的value和id的方法
今天小编就为大家分享一篇django获取from表单multiple-select的value和id的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
今天小编就为大家分享一篇python取均匀不重复的随机数方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-11-11
python装饰器就是用于拓展原来函数功能的一种函数，这个函数的特殊之处在于它的返回值也是一个函数，使用python装饰器的好处就是在不用更改原函数的代码前提下给函数增加新的功能2021-07-07
最新评论