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就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Python类的定义与使用，类名只要是一个合法的标识符即可，但这仅仅满足的是 Python 的语法要求：如果从程序的可读性方面来看，Python 的类名必须是由一个或多个有意义的单词连缀而成的，下文基于这些基础内容展开介绍,需要的朋友可以参考一下2022-03-03
这篇文章主要介绍了基于Python的一个自动录入表格的小程序，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-08-08
今天小编就为大家分享一篇java中两个byte数组实现合并的示例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
本次内容是小编在网上整理的关于如何python数据封装json格式的内容总结，有兴趣的读者们参考下。2018-03-03
集调试共享及成本控制Prompt工具PromptLayer使用指南
这篇文章主要介绍了集调试共享及成本控制Prompt工具PromptLayer使用指南,有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-03-03
这篇文章主要介绍了python ChainMap 合并字典的实现步骤，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-06-06
这篇文章主要介绍了Python入门之三角函数tan()的相关内容，介绍了tan（）函数的描述，语法以及简单实例，具有一定参考价值，需要的朋友可以了解下。2017-11-11
这篇文章主要介绍了在Python中使用NLTK库实现对词干的提取的教程,其中还用到了Pandas和IPython,需要的朋友可以参考下2015-04-04
这篇文章主要为大家介绍了Python深拷贝浅拷贝的图文示例清晰整理，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-06-06
这篇文章主要为大家详细介绍了Ubuntu18.04下python版本完美切换的解决方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-06-06
最新评论