Tensorflow 1.0之后模型文件、权重数值的读取方式
更新时间：2020年02月12日 08:41:12 作者：地大大刘
今天小编就为大家分享一篇Tensorflow 1.0之后模型文件、权重数值的读取方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
保存的文件有4个：
checkpoint
model-parameters.bin-46000.data-00000-of-00001
model-parameters.bin-46000.index
model-parameters.bin-46000.meta
读取代码：
from tensorflow.python import pywrap_tensorflow
with tf.Graph().as_default():
with tf.Session() as sess:
ckpt = tf.train.get_checkpoint_state('./checkpoint')
if ckpt and ckpt.model_checkpoint_path:
reader = pywrap_tensorflow.NewCheckpointReader('./checkpoint/model-parameters.bin-46000')
all_variables = reader.get_variable_to_shape_map()
w1 = reader.get_tensor("conv1/weight")
print(w1.shape)
print(w1)
print(all_variables)
else:
print('No checkpoint file found')
以上这篇Tensorflow 1.0之后模型文件、权重数值的读取方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇TFRecord格式存储数据与队列读取实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
这篇文章主要介绍了Python项目跨域问题解决方案,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-06-06
这篇文章主要介绍了一文详述 Python 中的 property 语法,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-09-09
今天小编就为大家分享一篇python3中的md5加密实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
这篇文章主要给大家介绍了关于python3对拉勾数据进行可视化分析的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用Python3具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧2019-04-04
Dataframe是Python中一种重要的数据处理工具,它能够以表格形式存储并处理数据,本文主要介绍了Python读取DataFrame的某行或某列的方法实现,具有一定的参考价值,感兴趣的可以了解一下2024-04-04
python微信聊天机器人改进版(定时或触发抓取天气预报、励志语录等,向好友推送)
这篇文章主要介绍了python微信聊天机器人改进版，定时或触发抓取天气预报、励志语录等，向好友推送 ,本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下2019-04-04
这篇文章主要介绍了python实现屏保计时器的示例代码，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-08-08
这篇文章主要为大家详细介绍了Python tkinter实现日期选择器，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-02-02
这篇文章主要介绍了python中numpy常用操作总结，NumPy是Python语言的一个扩充程序库，支持大量高维度数组与矩阵运算，此外也针对数组运算提供大量的数学函数库2022-09-09
最新评论