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实现的百度贴吧网络爬虫实例
Apr 17 Python
Python探索之静态方法和类方法的区别详解
Oct 27 Python
用PyInstaller把Python代码打包成单个独立的exe可执行文件
May 26 Python
Python实现获取邮箱内容并解析的方法示例
Jun 16 Python
python实现机器学习之元线性回归
Sep 06 Python
详解Numpy中的数组拼接、合并操作(concatenate, append, stack, hstack, vstack, r_, c_等)
May 27 Python
使用Python OpenCV为CNN增加图像样本的实现
Jun 10 Python
Python实现大数据收集至excel的思路详解
Jan 03 Python
Python 读取WAV音频文件 画频谱的实例
Mar 14 Python
python3 正则表达式基础廖雪峰
Mar 25 Python
virtualenv介绍及简明教程
Jun 23 Python
在pycharm中无法import所安装的库解决方案
May 31 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一些常用的正则表达式字符的一些转换
2008/07/29 PHP
php摘要生成函数(无乱码)
2012/02/04 PHP
解析Ubuntu下crontab命令的用法
2013/06/24 PHP
Riot.js 快速的JavaScript单元测试框架
2009/11/09 Javascript
formValidator3.3的ajaxValidator一些异常分析
2011/07/12 Javascript
探讨JavaScript中声明全局变量三种方式的异同
2013/12/03 Javascript
jQuery插件StickUp实现网页导航置顶
2015/04/12 Javascript
jquery实现顶部向右伸缩的导航区域代码
2015/09/02 Javascript
谈谈Jquery中的children find 的区别有哪些
2015/10/19 Javascript
React中获取数据的3种方法及优缺点
2020/02/18 Javascript
python实现获取客户机上指定文件并传输到服务器的方法
2015/03/16 Python
Python2和Python3中print的用法示例总结
2017/10/25 Python
浅谈Python实现Apriori算法介绍
2017/12/20 Python
解决phantomjs截图失败,phantom.exit位置的问题
2018/05/17 Python
Django框架自定义session处理操作示例
2019/05/27 Python
利用python实现冒泡排序算法实例代码
2019/12/01 Python
关于win10在tensorflow的安装及在pycharm中运行步骤详解
2020/03/16 Python
Python多个装饰器的调用顺序实例解析
2020/05/22 Python
解决TensorFlow程序无限制占用GPU的方法
2020/06/30 Python
pytorch随机采样操作SubsetRandomSampler()
2020/07/07 Python
python 中关于pycharm选择运行环境的问题
2020/10/31 Python
sklearn中的交叉验证的实现(Cross-Validation)
2021/02/22 Python
HTML5新标签兼容——> 的两种方法
2018/09/12 HTML / CSS
amazeui 验证按钮扩展的实现
2020/08/21 HTML / CSS
idealfit英国:世界领先的女性健身用品和运动衣物品牌
2017/11/25 全球购物
世界上最伟大的马产品:Equiderma
2020/01/07 全球购物
企业为何需要商业计划书
2013/12/26 职场文书
精神文明单位申报材料
2014/05/02 职场文书
领导班子对照检查材料
2014/09/22 职场文书
交通事故一次性赔偿协议书范本
2014/11/02 职场文书
2014年小学教学工作总结
2014/11/13 职场文书
物业公司管理制度
2015/08/05 职场文书
Python制作一个随机抽奖小工具的实现
2021/07/07 Python
AJAX实现省市县三级联动效果
2021/10/16 Javascript
一次Mysql update sql不当引起的生产故障记录
2022/04/01 MySQL
Win11任务栏无法正常显示 资源管理器不停重启的解决方法
2022/07/07 数码科技