tensorflow 实现从checkpoint中获取graph信息


Posted in Python onFebruary 10, 2020

代码:

import tensorflow as tf
 
sess = tf.Session()
check_point_path = 'variables' 
saver = tf.train.import_meta_graph('variables/save_variables.ckpt.meta')
 
saver.restore(sess, tf.train.latest_checkpoint(check_point_path))
 
graph = tf.get_default_graph()
 
#print(graph.get_operations())
 
#with open('op.txt','a') as f:
# f.write(str(graph.get_operations()))
op1 = graph.get_tensor_by_name('fully_connected/biases:0')
print(op1)

使用函数graph.get_operations()获取ckpt.meta中保存的graph中的所有operation,而tensor_name为'op_name:0'。

然后使用graph.get_tensor_by_name('op_name:0') 获取tensor信息。

代码从ckpt文件中获取保存的variable的数据(tensor的name和value):

import os
import tensorflow as tf
from tensorflow.python import pywrap_tensorflow
check_point_path = 'variables'
#checkpoint_path = os.path.join(logs_train_dir, 'model.ckpt')
ckpt = tf.train.get_checkpoint_state(checkpoint_dir=check_point_path)
checkpoint_path = os.path.join('.', ckpt.model_checkpoint_path)
#print(ckpt.model_checkpoint_path)
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
 print("tensor_name: ", key)
 #print(reader.get_tensor(key))

法二:

from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file
 
print_tensors_in_checkpoint_file("variables/save_variables.ckpt",tensor_name='', all_tensors=False, all_tensor_names=False)

注意:tf.train.latest_checkpoint(check_point_path) 方法用来获取最后一次ckeckpoint的路径,等价于

ckpt = tf.train.get_checkpoint_state(check_point_path)
ckpt.model_checkpoint_path

不能将tf.train.latest_checkpoint与tf.train.get_checkpoint_state 搞混了

以上这篇tensorflow 实现从checkpoint中获取graph信息就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
用python + openpyxl处理excel2007文档思路以及心得
Jul 14 Python
Python的Django框架中的URL配置与松耦合
Jul 15 Python
详解Django框架中的视图级缓存
Jul 23 Python
Python中map,reduce,filter和sorted函数的使用方法
Aug 17 Python
利用Python3分析sitemap.xml并抓取导出全站链接详解
Jul 04 Python
单利模式及python实现方式详解
Mar 20 Python
python 应用之Pycharm 新建模板默认添加编码格式-作者-时间等信息【推荐】
Jun 17 Python
Django Rest framework认证组件详细用法
Jul 25 Python
Selenium向iframe富文本框输入内容过程图解
Apr 10 Python
Python Dataframe常见索引方式详解
May 27 Python
Python读写csv文件流程及异常解决
Oct 20 Python
只用Python就可以制作的简单词云
Jun 07 Python
Python3 集合set入门基础
Feb 10 #Python
Django的CVB实例详解
Feb 10 #Python
TensorFlow实现checkpoint文件转换为pb文件
Feb 10 #Python
Django关于admin的使用技巧和知识点
Feb 10 #Python
Python实现括号匹配方法详解
Feb 10 #Python
Python re正则表达式元字符分组()用法分享
Feb 10 #Python
Tensorflow: 从checkpoint文件中读取tensor方式
Feb 10 #Python
You might like
php中3种方法统计字符串中每种字符的个数并排序
2012/08/27 PHP
PHP下使用CURL方式POST数据至API接口的代码
2013/02/14 PHP
如何使用php判断服务器是否是HTTPS连接
2013/07/05 PHP
php把大写命名转换成下划线分割命名
2015/04/27 PHP
Apache连接PHP后无法启动问题解决思路
2015/06/18 PHP
PHP单例模式详解及实例代码
2016/12/21 PHP
PHP封装的非对称加密RSA算法示例
2018/05/28 PHP
PHP变量的作用范围实例讲解
2020/12/22 PHP
Extjs中ComboBox加载并赋初值的实现方法
2012/03/22 Javascript
js为空或不是对象问题的快速解决方法
2013/12/11 Javascript
js之ActiveX控件使用说明 new ActiveXObject()
2014/03/03 Javascript
Jquery通过JSON字符串创建JSON对象
2014/08/24 Javascript
JS获取当前脚本文件的绝对路径
2016/03/02 Javascript
微信开发 使用picker封装省市区三级联动模板
2016/10/28 Javascript
vue中七牛插件使用的实例代码
2017/07/28 Javascript
Vue.js 动态为img的src赋值方法
2018/03/14 Javascript
vue改变对象或数组时的刷新机制的方法总结
2019/04/24 Javascript
使用Vue.set()方法实现响应式修改数组数据步骤
2019/11/09 Javascript
Vue引入Stylus知识点总结
2020/01/16 Javascript
详解搭建一个vue-cli的移动端H5开发模板
2020/01/17 Javascript
JavaScript中继承原理与用法实例入门
2020/05/09 Javascript
js+canvas实现刮刮奖功能
2020/09/13 Javascript
jQuery插件实现图片轮播效果
2020/10/19 jQuery
JavaScript 获取滚动条位置并将页面滑动到锚点
2021/02/08 Javascript
python分析nignx访问日志脚本分享
2015/02/26 Python
django进阶之cookie和session的使用示例
2018/08/17 Python
python 求某条线上特定x值或y值的点坐标方法
2019/07/09 Python
pycharm激活码快速激活及使用步骤
2020/03/12 Python
python推导式的使用方法实例
2021/02/28 Python
HTML5的结构和语义(5):内嵌媒体
2008/10/17 HTML / CSS
Otticanet美国:最顶尖的世界名牌眼镜, 能得到打折季的价格
2019/03/10 全球购物
解释一下抽象方法和抽象类
2016/08/27 面试题
授权委托书(法人单位用)
2014/09/29 职场文书
关于教师节的广播稿
2015/08/19 职场文书
毕业生求职自荐信(2016最新版)
2016/01/28 职场文书
一篇文章弄懂Python关键字、标识符和变量
2021/07/15 Python