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数据结构中字典的心得
Dec 04 Python
Python实现自动发送邮件功能
Mar 02 Python
Python学生信息管理系统修改版
Mar 13 Python
Python3随机漫步生成数据并绘制
Aug 27 Python
python 找出list中最大或者最小几个数的索引方法
Oct 30 Python
Python检查和同步本地时间(北京时间)的实现方法
Dec 03 Python
python导入pandas具体步骤方法
Jun 23 Python
django admin 自定义替换change页面模板的方法
Aug 23 Python
关于python 的legend图例,参数使用说明
Apr 17 Python
使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)
Jun 30 Python
Python可变集合和不可变集合的构造方法大全
Dec 06 Python
Python socket如何解析HTTP请求内容
Feb 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中register_globals参数为OFF和ON的区别(register_globals 使用详解)
2012/02/05 PHP
php的一个简单加密解密代码
2014/01/14 PHP
PHP下载远程文件到本地存储的方法
2015/03/24 PHP
如何把php5.3版本升级到php5.4或者php5.5
2015/07/31 PHP
highchart数据源纵轴json内的值必须是int(详解)
2017/02/20 PHP
利用PHP判断是手机移动端还是PC端访问的函数示例
2017/12/14 PHP
JQuery操作iframe父页面与子页面的元素与方法(实例讲解)
2013/11/20 Javascript
JS比较两个时间大小的简单示例代码
2013/12/20 Javascript
ECMAScript5中的对象存取器属性:getter和setter介绍
2014/12/08 Javascript
JavaScript中常见的字符串操作函数及用法汇总
2015/05/04 Javascript
深入理解jQuery layui分页控件的使用
2016/08/17 Javascript
js控住DOM实现发布微博效果
2016/08/30 Javascript
AngularJS通过$http和服务器通信详解
2016/09/21 Javascript
使用开源工具制作网页验证码的方法
2016/10/17 Javascript
详解微信小程序——自定义圆形进度条
2016/12/29 Javascript
JavaScript实现类似淘宝的购物车效果
2017/03/16 Javascript
clipboard在vue中的使用的方法示例
2018/10/19 Javascript
详解jquery和vue对比
2019/04/16 jQuery
ES6 Set结构的应用实例分析
2019/06/26 Javascript
vue使用require.context实现动态注册路由
2020/12/25 Vue.js
python opencv实现图片旋转矩形分割
2018/07/26 Python
Python高级特性切片(Slice)操作详解
2018/09/27 Python
对Python 窗体(tkinter)树状数据(Treeview)详解
2018/10/11 Python
TensorFlow 显存使用机制详解
2020/02/03 Python
numpy的Fancy Indexing和array比较详解
2020/06/11 Python
Ubuntu20.04环境安装tensorflow2的方法步骤
2021/01/29 Python
世界上最大的在线汽车租赁预订平台:Rentalcars.com(支持中文)
2018/10/12 全球购物
美国专业消费电子及摄影器材网站:B&H Photo Video
2019/12/18 全球购物
翻译学院毕业生自荐书
2014/02/02 职场文书
优秀应届毕业生推荐信
2014/02/18 职场文书
党课培训主持词
2014/04/01 职场文书
会计毕业生自荐书
2014/06/12 职场文书
公务员培的训心得体会
2014/09/01 职场文书
个人授权委托书模板
2014/09/14 职场文书
公务员群众路线心得体会
2014/11/03 职场文书
nginx location中多个if里面proxy_pass的方法
2021/03/31 Servers