tensorflow 获取checkpoint中的变量列表实例


Posted in Python onFebruary 11, 2020

方式1:静态获取,通过直接解析checkpoint文件获取变量名及变量值

通过

reader = tf.train.NewCheckpointReader(model_path)

或者通过:

from tensorflow.python import pywrap_tensorflow
reader = pywrap_tensorflow.NewCheckpointReader(model_path)

代码:

model_path = "./checkpoints/model.ckpt-75000"
## 下面两个reader作用等价
#reader = pywrap_tensorflow.NewCheckpointReader(model_path)
reader = tf.train.NewCheckpointReader(model_path)
 
## 用reader获取变量字典,key是变量名,value是变量的shape
var_to_shape_map = reader.get_variable_to_shape_map()
for var_name in var_to_shape_map.keys():
 #用reader获取变量值
 var_value = reader.get_tensor(var_name)
 
 print("var_name",var_name)
 print("var_value",var_value)

方式2:动态获取,先加载checkpoint模型,然后用graph.get_tensor_by_name()获取变量值

代码 (注意:要先在脚本中构建model中对应的变量及scope):

model_path = "./checkpoints/model.ckpt-75000"
 config = tf.ConfigProto()
 config.gpu_options.allow_growth = True
 with tf.Session(config=config) as sess:
  ## 获取待加载的变量列表
  trainable_vars = tf.trainable_variables()
  g_vars = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES,scope="generator")
  d_vars = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES,scope='discriminator')
  flow_vars = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES,scope='flow_net')
  var_restore = g_vars + d_vars
 
  ## 仅加载目标变量
  loader = tf.train.Saver(var_restore)
  loader.restore(sess,model_path)
 
  ## 显示加载的变量值
  graph = tf.get_default_graph()
  for var in var_restore:
   tensor = graph.get_tensor_by_name(var.name)
   print("=======变量名=======",tensor)
   print("-------变量值-------",sess.run(tensor))

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

Python 相关文章推荐
基于Python实现对PDF文件的OCR识别
Aug 05 Python
python安装PIL模块时Unable to find vcvarsall.bat错误的解决方法
Sep 19 Python
python opencv 图像尺寸变换方法
Apr 02 Python
django使用haystack调用Elasticsearch实现索引搜索
Jul 24 Python
pywinauto自动化操作记事本
Aug 26 Python
python主线程与子线程的结束顺序实例解析
Dec 17 Python
Python 模拟动态产生字母验证码图片功能
Dec 24 Python
TensorFLow 不同大小图片的TFrecords存取实例
Jan 20 Python
查看jupyter notebook每个单元格运行时间实例
Apr 22 Python
Python多线程threading创建及使用方法解析
Jun 17 Python
python修改微信和支付宝步数的示例代码
Oct 12 Python
python数字图像处理之图像的批量处理
Jun 28 Python
python使用正则表达式去除中文文本多余空格,保留英文之间空格方法详解
Feb 11 #Python
python 函数中的参数类型
Feb 11 #Python
python正则过滤字母、中文、数字及特殊字符方法详解
Feb 11 #Python
python3正则模块re的使用方法详解
Feb 11 #Python
Python版中国省市经纬度
Feb 11 #Python
Python终端输出彩色字符方法详解
Feb 11 #Python
Python连接Oracle之环境配置、实例代码及报错解决方法详解
Feb 11 #Python
You might like
PHP分页显示制作详细讲解
2008/11/19 PHP
PHP中include()与require()的区别说明
2010/03/10 PHP
php function用法如何递归及return和echo区别
2014/03/07 PHP
CI框架给视图添加动态数据
2014/12/01 PHP
PHP中使用Memache作为进程锁的操作类分享
2015/03/30 PHP
PHP 8新特性简介
2020/08/18 PHP
一个XML格式数据转换为图表的例子
2010/02/09 Javascript
jQuery-Easyui 1.2 实现多层菜单效果的代码
2012/01/13 Javascript
javascript页面渲染速度测试脚本分享
2014/04/15 Javascript
每天一篇javascript学习小结(Array数组)
2015/11/11 Javascript
jquery实现图片预加载
2015/12/25 Javascript
JavaScript SHA1加密算法实现详细代码
2016/10/06 Javascript
微信小程序 摇一摇抽奖简单实例实现代码
2017/01/09 Javascript
Bootstarp基本模版学习教程
2017/02/01 Javascript
jQuery实现返回顶部按钮和scroll滚动功能[带动画效果]
2017/07/05 jQuery
详解使用Node.js 将txt文件转为Excel文件
2017/07/05 Javascript
Nuxt.js 数据双向绑定的实现
2019/02/17 Javascript
JS重学系列之聊聊new操作符
2019/03/04 Javascript
js实现数字从零慢慢增加到指定数字示例
2019/11/07 Javascript
three.js 如何制作魔方
2020/07/31 Javascript
python实现的文件夹清理程序分享
2014/11/22 Python
python基于ID3思想的决策树
2018/01/03 Python
python学生信息管理系统(初级版)
2018/10/17 Python
python实现五子棋小程序
2019/06/18 Python
python中PS 图像调整算法原理之亮度调整
2019/06/28 Python
Python学习笔记之For循环用法详解
2019/08/14 Python
Python绘制股票移动均线的实例
2019/08/24 Python
基于python的selenium两种文件上传操作实现详解
2019/09/19 Python
粉红色的鲸鱼:Vineyard Vines
2018/02/17 全球购物
Jdbc数据访问技术面试题
2012/03/30 面试题
不忘国耻振兴中华演讲稿
2014/05/14 职场文书
人口与计划生育目标管理责任书
2014/07/29 职场文书
2014光棍节单身联谊活动策划书
2014/10/10 职场文书
教师年终个人总结
2015/02/11 职场文书
小学英语课教学反思
2016/02/15 职场文书
tensorflow中的数据类型dtype用法说明
2021/05/26 Python