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获取从命令行输入数字的方法
Apr 29 Python
Python实现导出数据生成excel报表的方法示例
Jul 12 Python
matplotlib中legend位置调整解析
Dec 19 Python
scrapy-redis源码分析之发送POST请求详解
May 15 Python
Python缓存技术实现过程详解
Sep 25 Python
详解Python time库的使用
Oct 10 Python
django框架单表操作之增删改实例分析
Dec 16 Python
Python序列类型的打包和解包实例
Dec 21 Python
Python爬虫如何应对Cloudflare邮箱加密
Jun 24 Python
python脚本使用阿里云slb对恶意攻击进行封堵的实现
Feb 04 Python
python 遍历磁盘目录的三种方法
Apr 02 Python
python实现求纯色彩图像的边框
Apr 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实现移除数组中为空或为某值元素的方法
2017/01/07 PHP
php基于PDO实现功能强大的MYSQL封装类实例
2017/02/27 PHP
TP5多入口设置实例讲解
2020/12/15 PHP
jQuery Ajax提交表单查询获得数据实例代码
2012/09/19 Javascript
一行命令搞定node.js 版本升级
2014/07/20 Javascript
JS获得选取checkbox整行数据的方法
2015/01/28 Javascript
js封装tab标签页实例分享
2016/12/19 Javascript
js实现textarea限制输入字数
2017/02/13 Javascript
jQuery使用EasyUi实现三级联动下拉框效果
2017/03/08 Javascript
protractor的安装与基本使用教程
2017/07/07 Javascript
深入理解ES7的async/await的用法
2017/09/09 Javascript
jquery 获取索引值在一定范围的列表方法
2018/01/25 jQuery
vue+elementUI实现表格关键字筛选高亮
2020/10/26 Javascript
关于JSON解析的实现过程解析
2019/10/08 Javascript
JS实现碰撞检测效果
2020/03/12 Javascript
[01:08:44]NB vs VP 2018国际邀请赛小组赛BO2 第一场 8.18
2018/08/19 DOTA
python中查找excel某一列的重复数据 剔除之后打印
2013/02/10 Python
极简的Python入门指引
2015/04/01 Python
python访问类中docstring注释的实现方法
2015/05/04 Python
Python IDE PyCharm的基本快捷键和配置简介
2015/11/04 Python
Python的Flask框架的简介和安装方法
2015/11/13 Python
pycharm下打开、执行并调试scrapy爬虫程序的方法
2017/11/29 Python
详解python校验SQL脚本命名规则
2019/03/22 Python
pycharm设置鼠标悬停查看方法设置
2019/07/29 Python
python实现单链表的方法示例
2019/09/03 Python
python TK库简单应用(实时显示子进程输出)
2019/10/29 Python
通过代码实例了解Python sys模块
2020/09/14 Python
经典C++面试题一
2016/11/06 面试题
当我正在为表建立索引的时候,SQL Server 会禁止对表的访问吗
2014/04/28 面试题
新锐科技Java程序员面试题
2016/07/25 面试题
社区学习十八大感想
2014/01/22 职场文书
学生安全教育材料
2014/02/14 职场文书
师范生自我鉴定
2014/03/20 职场文书
公司门卫岗位职责
2015/04/13 职场文书
MySQL基础(二)
2021/04/05 MySQL
Spring Cloud OAuth2实现自定义token返回格式
2022/06/25 Java/Android