Tensorflow 1.0之后模型文件、权重数值的读取方式


Posted in Python onFebruary 12, 2020

保存的文件有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之后模型文件、权重数值的读取方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
在漏洞利用Python代码真的很爽
Aug 26 Python
Python发送email的3种方法
Apr 28 Python
Python基于checksum计算文件是否相同的方法
Jul 09 Python
实践Python的爬虫框架Scrapy来抓取豆瓣电影TOP250
Jan 20 Python
Python工厂函数用法实例分析
May 14 Python
用python 实现在不确定行数情况下多行输入方法
Jan 28 Python
利用python求积分的实例
Jul 03 Python
Python 合并多个TXT文件并统计词频的实现
Aug 23 Python
python实现高斯(Gauss)迭代法的例子
Nov 20 Python
如何在mac环境中用python处理protobuf
Dec 25 Python
关于Tensorflow 模型持久化详解
Feb 12 Python
python numpy库np.percentile用法说明
Jun 08 Python
Python django框架开发发布会签到系统(web开发)
Feb 12 #Python
Python计算公交发车时间的完整代码
Feb 12 #Python
详解Django3中直接添加Websockets方式
Feb 12 #Python
Tensorflow 使用pb文件保存(恢复)模型计算图和参数实例详解
Feb 11 #Python
TensorFlow:将ckpt文件固化成pb文件教程
Feb 11 #Python
TensorFlow获取加载模型中的全部张量名称代码
Feb 11 #Python
tensorflow 获取checkpoint中的变量列表实例
Feb 11 #Python
You might like
PHP面向对象法则
2012/02/23 PHP
3个PHP多维数组转为一维数组的方法实例
2014/03/13 PHP
php开发工具有哪五款
2015/11/09 PHP
[原创]PHP简单开启curl的方法(测试可行)
2016/01/11 PHP
Laravel框架下的Contracts契约详解
2020/03/17 PHP
禁止IE用右键的JS代码
2013/12/30 Javascript
jquery trigger伪造a标签的click事件取代window.open方法
2014/06/23 Javascript
jquery实现显示已选用户
2014/07/21 Javascript
Backbone.js中的集合详解
2015/01/14 Javascript
JavaScript用select实现日期控件
2015/07/17 Javascript
JS实现的幻灯片切换显示效果
2016/09/07 Javascript
JS+HTML5 FileReader实现文件上传前本地预览功能
2020/03/27 Javascript
easyui-datagrid开发实践(总结)
2017/08/02 Javascript
在Vue中使用icon 字体图标的方法
2019/06/14 Javascript
js基础之事件捕获与冒泡原理
2019/10/09 Javascript
vue.js iview打包上线后字体图标不显示解决办法
2020/01/20 Javascript
python关闭windows进程的方法
2015/04/18 Python
浅谈PyTorch的可重复性问题(如何使实验结果可复现)
2020/02/20 Python
Python3如何在Windows和Linux上打包
2020/02/25 Python
keras 简单 lstm实例(基于one-hot编码)
2020/07/02 Python
Python脚本实现Zabbix多行日志监控过程解析
2020/08/26 Python
35款精致的 CSS3 和 HTML5 网页模板 推荐
2012/08/03 HTML / CSS
利用CSS3 动画 绘画 圆形动态时钟
2018/03/20 HTML / CSS
Expedia意大利旅游网站:酒店、机票和租车预订
2017/10/30 全球购物
Goodee官方商店:迷你投影仪
2021/03/15 全球购物
编辑找工作求职信分享
2014/01/03 职场文书
护理人员的自我评价分享
2014/03/15 职场文书
班组拓展活动方案
2014/08/14 职场文书
学校联谊协议书
2014/09/16 职场文书
公司授权委托书格式样本
2014/10/01 职场文书
小区环境卫生倡议书
2015/04/29 职场文书
2015年污水处理厂工作总结
2015/05/26 职场文书
复兴之路纪录片观后感
2015/06/02 职场文书
社会实践心得体会范文
2016/01/14 职场文书
Go语言并发编程 sync.Once
2021/10/16 Golang
全新239军机修复记
2022/04/05 无线电