tensorflow实现读取模型中保存的值 tf.train.NewCheckpointReader


Posted in Python onFebruary 10, 2020

使用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"))

tensorflow实现读取模型中保存的值 tf.train.NewCheckpointReader

其中有3个字段, 分别是名字, 数据类型, shape

获取变量中的值

print("get_tensor:\n")
pprint.pprint(NewCheck.get_tensor("D/conv2d/bias"))

tensorflow实现读取模型中保存的值 tf.train.NewCheckpointReader

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

以上这篇tensorflow实现读取模型中保存的值 tf.train.NewCheckpointReader就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中使用PIPE操作Linux管道
Feb 04 Python
Python的条件语句与运算符优先级详解
Oct 13 Python
python黑魔法之参数传递
Feb 12 Python
linux查找当前python解释器的位置方法
Feb 20 Python
Python函数基础实例详解【函数嵌套,命名空间,函数对象,闭包函数等】
Mar 30 Python
Python实现的多进程拷贝文件并显示百分比功能示例
Apr 09 Python
使用Django开发简单接口实现文章增删改查
May 09 Python
详解pandas DataFrame的查询方法(loc,iloc,at,iat,ix的用法和区别)
Aug 02 Python
flask 实现上传图片并缩放作为头像的例子
Jan 09 Python
PyQt5中QTableWidget如何弹出菜单的示例代码
Feb 23 Python
Jupyter notebook运行Spark+Scala教程
Apr 10 Python
python代码中怎么换行
Jun 17 Python
解决tensorflow添加ptb库的问题
Feb 10 #Python
使用TensorFlow直接获取处理MNIST数据方式
Feb 10 #Python
python如何实现单链表的反转
Feb 10 #Python
基于python求两个列表的并集.交集.差集
Feb 10 #Python
python实现串口通信的示例代码
Feb 10 #Python
基于Tensorflow高阶读写教程
Feb 10 #Python
python集合删除多种方法详解
Feb 10 #Python
You might like
windows下开发并编译PHP扩展的方法
2011/03/18 PHP
PHP5中新增stdClass 内部保留类
2011/06/13 PHP
PHP容易被忽略而出错陷阱 数字与字符串比较
2011/11/10 PHP
PHP获取本周第一天和最后一天示例代码
2014/02/24 PHP
PHP写日志的实现方法
2014/11/05 PHP
php+mysql查询优化简单实例
2015/01/13 PHP
php准确获取文件MIME类型的方法
2015/06/17 PHP
纯php生成随机密码
2015/10/30 PHP
PHP与以太坊交互详解
2018/08/24 PHP
ExtJs扩展之GroupPropertyGrid代码
2010/03/05 Javascript
javascript向flash swf文件传递参数值注意细节
2012/12/11 Javascript
使用jquery自定义鼠标样式满足个性需求
2013/11/05 Javascript
jquery如何获取复选框的值
2013/12/12 Javascript
一个网页标题title的闪动提示效果实现思路
2014/03/22 Javascript
微信小程序 WXDropDownMenu组件详解及实例代码
2016/10/24 Javascript
javascript九宫格图片随机打乱位置的实现方法
2017/03/15 Javascript
mint-ui 时间插件使用及获取选择值的方法
2018/02/09 Javascript
详解微信小程序-获取用户session_key,openid,unionid - 后端为nodejs
2019/04/29 NodeJs
vuex存值与取值的实例
2019/11/06 Javascript
Postman内建变量常用方法实例解析
2020/07/28 Javascript
详谈Vue.js框架下main.js,App.vue,page/index.vue之间的区别
2020/08/12 Javascript
[46:50]Liquid vs Mineski 2018国际邀请赛小组赛BO2 第二场 8.18
2018/08/19 DOTA
Python基于sftp及rsa密匙实现远程拷贝文件的方法
2016/09/21 Python
python数据结构之链表的实例讲解
2017/07/25 Python
Python cookbook(数据结构与算法)实现对不原生支持比较操作的对象排序算法示例
2018/03/15 Python
Python使用matplotlib绘制三维图形示例
2018/08/25 Python
Python全面分析系统的时域特性和频率域特性
2020/02/26 Python
Python描述符descriptor使用原理解析
2020/03/21 Python
巴西补充剂和维生素购物网站:Natue
2019/06/17 全球购物
旷课检讨书大全
2014/01/21 职场文书
《猫》教学反思
2014/02/26 职场文书
法人代表授权委托书
2014/04/08 职场文书
2014年人力资源部工作总结
2014/11/19 职场文书
十月围城观后感
2015/06/08 职场文书
Python包argparse模块常用方法
2021/06/04 Python
ajax请求前端跨域问题原因及解决方案
2021/10/16 Javascript