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实现统计文本文件字数的方法
May 05 Python
Python yield与实现方法代码分析
Feb 06 Python
Python2中文处理纪要的实现方法
Mar 10 Python
python简单操作excle的方法
Sep 12 Python
win7 x64系统中安装Scrapy的方法
Nov 18 Python
Python告诉你木马程序的键盘记录原理
Feb 02 Python
python启动应用程序和终止应用程序的方法
Jun 28 Python
Python实现图片批量加入水印代码实例
Nov 30 Python
Python动态声明变量赋值代码实例
Dec 30 Python
使用tensorflow DataSet实现高效加载变长文本输入
Jan 20 Python
TensorFlow 输出checkpoint 中的变量名与变量值方式
Feb 11 Python
python+selenium小米商城红米K40手机自动抢购的示例代码
Mar 24 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读取mysql的简单实例
2014/01/15 PHP
php制作动态随机验证码
2015/02/12 PHP
基于OpenCart 开发支付宝,财付通,微信支付参数错误问题
2015/10/01 PHP
php数组指针操作详解
2017/02/14 PHP
php服务器的系统详解
2019/10/12 PHP
php中Swoole的热更新实现代码实例
2021/03/04 PHP
js实现权限树的更新权限时的全选全消功能
2009/02/17 Javascript
WordPress 照片lightbox效果的运用几点
2009/06/22 Javascript
腾讯UED 漂亮的提示信息效果代码
2011/09/12 Javascript
推荐40个简单的 jQuery 导航插件和教程(下篇)
2012/09/14 Javascript
JavaScript 模拟类机制及私有变量的方法及思路
2013/07/10 Javascript
jQuery插件分享之分页插件jqPagination
2014/06/06 Javascript
web前端设计师们常用的jQuery特效插件汇总
2014/12/07 Javascript
深入探寻seajs的模块化与加载方式
2015/04/14 Javascript
JavaScript模拟鼠标右键菜单效果
2020/12/08 Javascript
javascript中json基础知识详解
2017/01/19 Javascript
JavaScript运动框架 解决速度正负取整问题(一)
2017/05/17 Javascript
JavaScript寄生组合式继承原理与用法分析
2019/01/11 Javascript
python django集成cas验证系统
2014/07/14 Python
简单讲解Python中的数字类型及基本的数学计算
2016/03/11 Python
Python 装饰器使用详解
2017/07/29 Python
python利用MethodType绑定方法到类示例代码
2017/08/27 Python
python线程中同步锁详解
2018/04/27 Python
python 自动重连wifi windows的方法
2018/12/18 Python
python之验证码生成(gvcode与captcha)
2019/01/02 Python
重写django的model下的objects模型管理器方式
2020/05/15 Python
Vans(范斯)新西兰官方网站:美国原创极限运动品牌
2020/09/19 全球购物
大学毕业生通用求职信
2013/09/28 职场文书
给校长的建议书500字
2014/05/15 职场文书
党支部创先争优承诺书
2014/08/30 职场文书
教育局党的群众路线教育实践活动整改方案
2014/09/20 职场文书
党风廉正建设责任书
2015/01/29 职场文书
史上最牛的辞职信
2015/02/28 职场文书
2015年生产部工作总结范文
2015/05/25 职场文书
法律意见书范本
2015/06/04 职场文书
mybatis 获取无数据的字段不显示的问题
2021/07/15 Java/Android