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 相关文章推荐
50行代码实现贪吃蛇(具体思路及代码)
Apr 27 Python
python实现获取序列中最小的几个元素
Sep 25 Python
调试Python程序代码的几种方法总结
Apr 28 Python
python妹子图简单爬虫实例
Jul 07 Python
Python实现基本线性数据结构
Aug 22 Python
Selenium(Python web测试工具)基本用法详解
Aug 10 Python
Python考拉兹猜想输出序列代码实践
Jul 05 Python
Python + OpenCV 实现LBP特征提取的示例代码
Jul 11 Python
简单介绍python封装的基本知识
Aug 10 Python
pytorch GAN生成对抗网络实例
Jan 10 Python
Python‘==‘ 及 ‘is‘相关原理解析
Sep 05 Python
端午节将至,用Python爬取粽子数据并可视化,看看网友喜欢哪种粽子吧!
Jun 11 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
再说下636单管机
2021/03/02 无线电
php cache类代码(php数据缓存类)
2010/04/15 PHP
PHP随机数生成代码与使用实例分析
2011/04/08 PHP
PHP使用socket发送HTTP请求的方法
2016/02/14 PHP
PHP实现财务审核通过后返现金额到客户的功能
2019/07/04 PHP
PHP实现单条sql执行多个数据的insert语句方法
2019/10/11 PHP
PHP+MySQL实现在线测试答题实例
2020/01/02 PHP
获取dom元素那些讨厌的位置封装代码
2010/06/23 Javascript
js 在定义的时候立即执行的函数表达式(function)写法
2013/01/16 Javascript
JS和函数式语言的三特性
2014/03/05 Javascript
NodeJS学习笔记之Connect中间件模块(一)
2015/01/27 NodeJs
jscript读写二进制文件的方法
2015/04/22 Javascript
AngularJS 简单应用实例
2016/07/28 Javascript
JavaScript计时器用法分析【setTimeout和clearTimeout】
2017/01/18 Javascript
Angular.js中ng-if、ng-show和ng-hide的区别介绍
2017/01/20 Javascript
ES6新特性之函数的扩展实例详解
2017/04/01 Javascript
JavaScript转换数据库DateTime字段类型方法
2017/06/27 Javascript
浅谈Node异步编程的机制
2017/10/18 Javascript
vue项目中跳转到外部链接的实例讲解
2018/09/20 Javascript
Koa代理Http请求的示例代码
2018/10/10 Javascript
构建Vue大型应用的10个最佳实践(小结)
2019/11/07 Javascript
vue图片裁剪插件vue-cropper使用方法详解
2020/12/16 Vue.js
Python中的模块和包概念介绍
2015/04/13 Python
对Tensorflow中权值和feature map的可视化详解
2018/06/14 Python
详解Python的三种可变参数
2019/05/08 Python
Python3基础教程之递归函数简单示例
2019/06/07 Python
Python xpath表达式如何实现数据处理
2020/06/13 Python
中间件分为哪几类
2016/09/18 面试题
大学生社会实践评语
2014/04/25 职场文书
学校做一个有道德的人活动方案
2014/08/23 职场文书
大学团日活动新闻稿
2014/09/10 职场文书
领导干部群众路线剖析材料
2014/10/09 职场文书
毕业班班主任工作总结2015
2015/07/23 职场文书
《家》读后感:万惜拯救,冷暖自知
2019/09/25 职场文书
详解Mysql数据库平滑扩容解决高并发和大数据量问题
2022/05/25 MySQL
js作用域及作用域链工作引擎
2022/07/07 Javascript