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中使用异常处理来判断运行的操作系统平台方法
Jan 22 Python
在Python的Django框架中加载模版的方法
Jul 16 Python
python实现协同过滤推荐算法完整代码示例
Dec 15 Python
PYTHON基础-时间日期处理小结
May 05 Python
Python中psutil的介绍与用法
May 02 Python
Django框架模板文件使用及模板文件加载顺序分析
May 23 Python
pytorch使用 to 进行类型转换方式
Jan 08 Python
浅谈PyTorch的可重复性问题(如何使实验结果可复现)
Feb 20 Python
Python tcp传输代码实例解析
Mar 18 Python
Python HTMLTestRunner库安装过程解析
May 25 Python
pycharm中leetcode插件使用图文详解
Dec 07 Python
python drf各类组件的用法和作用
Jan 12 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生成excel列序号代码实例
2013/12/24 PHP
PHP扩展开发入门教程
2015/02/26 PHP
ThinkPHP下表单令牌错误与解决方法分析
2017/05/20 PHP
Nigma vs AM BO3 第一场2.13
2021/03/10 DOTA
web页面数据展示新想法(json)
2010/06/08 Javascript
javascript中的prototype属性实例分析说明
2010/08/09 Javascript
JavaScript中的私有/静态属性介绍
2012/07/26 Javascript
DOM基础教程之模型中的模型节点
2015/01/19 Javascript
jQuery使用before()和after()在元素前后添加内容的方法
2015/03/26 Javascript
jQuery form插件之ajaxForm()和ajaxSubmit()的可选参数项对象
2016/01/23 Javascript
Bootstrap学习笔记之css组件(3)
2016/06/07 Javascript
jqPlot jQuery绘图插件的使用
2016/06/18 Javascript
JS根据生日月份和日期计算星座的简单实现方法
2016/11/24 Javascript
JS实现自定义状态栏动画文字效果示例
2017/10/12 Javascript
仿淘宝JSsearch搜索下拉深度用法
2018/01/15 Javascript
JavaScript实现异步图像上传功能
2018/07/12 Javascript
AngularJS使用$http配置对象方式与服务端交互方法
2018/08/13 Javascript
js中null与空字符串""的区别讲解
2019/01/17 Javascript
js实现轮播图特效
2020/05/28 Javascript
Python2.5/2.6实用教程 入门基础篇
2009/11/29 Python
pyqt5 获取显示器的分辨率的方法
2019/06/18 Python
python根据多个文件名批量查找文件
2019/08/13 Python
softmax及python实现过程解析
2019/09/30 Python
Django多层嵌套ManyToMany字段ORM操作详解
2020/05/19 Python
社区版pycharm创建django项目的方法(pycharm的newproject左侧没有项目选项)
2020/09/23 Python
css3.0 图形构成实例练习一
2013/03/19 HTML / CSS
Swisse官方海外旗舰店:澳大利亚销量领先,自然健康品牌
2017/12/15 全球购物
俄罗斯第一家多品牌在线奢侈品精品店:Aizel.ru
2020/09/06 全球购物
市场营销专业个人自荐信格式
2013/09/21 职场文书
军神教学反思
2014/02/04 职场文书
小学校园之星事迹材料
2014/05/16 职场文书
民主评议政风行风活动心得体会
2014/10/29 职场文书
2015年女生节活动总结
2015/02/27 职场文书
三八节活动主持词
2015/07/04 职场文书
python编写函数注意事项总结
2021/03/29 Python
canvas多重阴影发光效果实现
2021/04/20 Javascript