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杀死一个线程的方法
Sep 06 Python
Python自动化运维_文件内容差异对比分析
Dec 13 Python
python使用phoenixdb操作hbase的方法示例
Feb 28 Python
python turtle库画一个方格和圆实例
Jun 27 Python
python挖矿算力测试程序详解
Jul 03 Python
简单了解python中的与或非运算
Sep 18 Python
tensorflow入门:TFRecordDataset变长数据的batch读取详解
Jan 20 Python
python seaborn heatmap可视化相关性矩阵实例
Jun 03 Python
python实现取余操作的简单实例
Aug 16 Python
python如何使用腾讯云发送短信
Sep 17 Python
使用pandas实现筛选出指定列值所对应的行
Dec 13 Python
python爬取豆瓣电影排行榜(requests)的示例代码
Feb 18 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 动态添加记录
2009/03/10 PHP
ThinkPHP、ZF2、Yaf、Laravel框架路由大比拼
2015/03/25 PHP
php实现断点续传大文件示例代码
2020/06/19 PHP
JQERY limittext 插件0.2版(长内容限制显示)
2010/08/27 Javascript
extjs之去除s.gif的影响
2010/12/25 Javascript
浅谈JavaScript编程语言的编码规范
2011/10/21 Javascript
jquery easyui滚动条部分设置介绍
2013/09/12 Javascript
js数组方法扩展实现数组统计函数
2014/04/09 Javascript
jQuery对于显示和隐藏等常用状态的判断方法
2014/12/13 Javascript
JavaScript 异常处理 详解
2015/02/06 Javascript
基于JavaScript实现仿京东图片轮播效果
2015/11/06 Javascript
jQuery实现简易的输入框字数计数功能示例
2017/01/16 Javascript
JavaScript数据结构之二叉树的遍历算法示例
2017/04/13 Javascript
Vue 父子组件数据传递的四种方式( inheritAttrs + $attrs + $listeners)
2018/05/04 Javascript
JS实现显示当前日期的实例代码
2018/07/03 Javascript
jQuery实现表单动态添加数据并提交的方法
2018/07/19 jQuery
微信小程序控制台提示warning:Now you can provide attr "wx:key" for a "wx:for" to improve performance解决方法
2019/02/21 Javascript
详解关于webpack多入口热加载很慢的原因
2019/04/24 Javascript
JavaScript之数组扁平化详解
2019/06/03 Javascript
Nuxt.js实现一个SSR的前端博客的示例代码
2019/09/06 Javascript
在vue中把含有html标签转为html渲染页面的实例
2019/10/28 Javascript
JavaScript实现京东放大镜效果
2019/12/03 Javascript
Vue使用鼠标在Canvas上绘制矩形
2020/12/24 Vue.js
[36:45]TNC vs VGJ.S 2018国际邀请赛小组赛BO2 第二场 8.18
2018/08/19 DOTA
python解决汉字编码问题:Unicode Decode Error
2017/01/19 Python
Python实现的径向基(RBF)神经网络示例
2018/02/06 Python
python地震数据可视化详解
2019/06/18 Python
大家都说好用的Python命令行库click的使用
2019/11/07 Python
python GUI库图形界面开发之PyQt5不规则窗口实现与显示GIF动画的详细方法与实例
2020/03/09 Python
网页布局中CSS样式无效的十个重要原因详解
2017/08/10 HTML / CSS
大学毕业生个人自荐信范文
2014/01/08 职场文书
就业协议书样本
2014/08/20 职场文书
预备党员2014年第四季度思想汇报范文
2014/10/25 职场文书
同意迁入证明模板
2014/10/26 职场文书
实习单位意见
2015/06/04 职场文书
mysql分组后合并显示一个字段的多条数据方式
2022/01/22 MySQL