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 相关文章推荐
在树莓派2或树莓派B+上安装Python和OpenCV的教程
Mar 30 Python
用Python的Django框架编写从Google Adsense中获得报表的应用
Apr 17 Python
Python编程修改MP3文件名称的方法
Apr 19 Python
Python实现的端口扫描功能示例
Apr 08 Python
使用pyinstaller打包PyQt4程序遇到的问题及解决方法
Jun 24 Python
pandas DataFrame 警告(SettingWithCopyWarning)的解决
Jul 23 Python
django基础学习之send_mail功能
Aug 07 Python
Python进程间通信 multiProcessing Queue队列实现详解
Sep 23 Python
python实现跨excel sheet复制代码实例
Mar 03 Python
Python如何批量获取文件夹的大小并保存
Mar 31 Python
如何实现更换Jupyter Notebook内核Python版本
May 18 Python
Python3的进程和线程你了解吗
Mar 16 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面向对象程序设计之命名空间与自动加载类详解
2016/12/02 PHP
Laravel5.7框架安装与使用学习笔记图文详解
2019/04/02 PHP
解决laravel id非自增 模型取回为0 的问题
2019/10/11 PHP
写入cookie的JavaScript代码库 cookieLibrary.js
2009/10/24 Javascript
jquery.validate使用攻略 第一部
2010/07/01 Javascript
jquery jqPlot API 中文使用教程(非常强大的图表工具)
2011/08/15 Javascript
使用jQuery validate 验证注册表单实例演示
2013/03/25 Javascript
javascript实现阻止iOS APP中的链接打开Safari浏览器
2014/06/12 Javascript
jQuery实现响应鼠标背景变化的动态菜单效果代码
2015/08/27 Javascript
JS组件Form表单验证神器BootstrapValidator
2016/01/26 Javascript
js拖拽功能实现代码解析
2016/11/28 Javascript
Vue中用props给data赋初始值遇到的问题解决
2018/11/27 Javascript
Python实现批量修改图片格式和大小的方法【opencv库与PIL库】
2018/12/03 Python
wxPython+Matplotlib绘制折线图表
2019/11/19 Python
使用Python实现Wake On Lan远程开机功能
2020/01/22 Python
pytorch使用tensorboardX进行loss可视化实例
2020/02/24 Python
浅谈django channels 路由误导
2020/05/28 Python
python能做哪些生活有趣的事情
2020/09/09 Python
Django数据统计功能count()的使用
2020/11/30 Python
html5录音功能实战示例
2019/03/25 HTML / CSS
西班牙网上书店:Casa del Libro
2016/11/01 全球购物
加拿大百叶窗和窗帘定制网站:Blinds
2017/01/30 全球购物
加拿大高尔夫超市:Golf Town
2018/01/12 全球购物
英国汽车零件购物网站:GSF Car Parts
2019/05/23 全球购物
C#公司笔试题
2014/03/28 面试题
医药大学生求职简历的自我评价
2013/10/17 职场文书
应届生财务管理求职信
2013/11/06 职场文书
财务会计毕业生个人求职信
2014/02/03 职场文书
新疆民族团结演讲稿
2014/08/27 职场文书
班子个人四风问题整改措施
2014/10/04 职场文书
委托证明范本
2014/11/25 职场文书
毕业论文答辩开场白
2015/05/27 职场文书
欠款证明
2015/06/24 职场文书
2016年清明节期间群众祭祀活动工作总结
2016/04/01 职场文书
JavaScript高级程序设计之基本引用类型
2021/11/17 Javascript
Go中使用gjson来操作JSON数据的实现
2022/08/14 Golang