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的Django框架部署的一些建议
Apr 09 Python
python学习教程之Numpy和Pandas的使用
Sep 11 Python
python简单商城购物车实例代码
Mar 15 Python
pandas数据集的端到端处理
Feb 18 Python
pyqt5实现登录界面的模板
May 30 Python
查看Python依赖包及其版本号信息的方法
Aug 13 Python
django drf框架自带的路由及最简化的视图
Sep 10 Python
PyQt5多线程刷新界面防假死示例
Dec 13 Python
python 爬取古诗文存入mysql数据库的方法
Jan 08 Python
Python实现在Windows平台修改文件属性
Mar 05 Python
Pytorch 解决自定义子Module .cuda() tensor失败的问题
Jun 23 Python
Python3爬虫ChromeDriver的安装实例
Feb 06 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功能函数
2015/04/15 PHP
用JTrackBar实现的模拟苹果风格的滚动条
2007/08/06 Javascript
JavaScript及jquey实现多个数组的合并操作
2014/09/06 Javascript
Shell脚本实现Linux系统和进程资源监控
2015/03/05 Javascript
JQuery调用绑定click事件的3种写法
2015/03/28 Javascript
基于javascript的异步编程实例详解
2017/04/10 Javascript
javaScript 逻辑运算符使用技巧整理
2017/05/03 Javascript
JS实现自动轮播图效果(自适应屏幕宽度+手机触屏滑动)
2017/06/19 Javascript
在Vue.js中使用Mixins的方法
2017/09/12 Javascript
angular4实现tab栏切换的方法示例
2017/10/21 Javascript
利用JS判断客户端类型你应该知道的四种方法
2017/12/22 Javascript
vue实现组件之间传值功能示例
2018/07/13 Javascript
vue做移动端适配最佳解决方案(亲测有效)
2018/09/04 Javascript
小程序分享模块超级详解(推荐)
2019/04/10 Javascript
angular 服务随记小结
2019/05/06 Javascript
layui 地区三级联动 form select 渲染的实例
2019/09/27 Javascript
python list中append()与extend()用法分享
2013/03/24 Python
使用Python编写简单网络爬虫抓取视频下载资源
2014/11/04 Python
python面向对象_详谈类的继承与方法的重载
2017/06/07 Python
基于Python代码编辑器的选用(详解)
2017/09/13 Python
使用python生成目录树
2018/03/29 Python
Python查找文件中包含中文的行方法
2018/12/19 Python
Python 图像处理: 生成二维高斯分布蒙版的实例
2019/07/04 Python
关于Python 的简单栅格图像边界提取方法
2019/07/05 Python
HTML5的表单(绝对特别强大的功能)使用示例
2013/06/20 HTML / CSS
施华洛世奇美国官网:SWAROVSKI美国
2018/02/08 全球购物
SIDESTEP荷兰:在线购买鞋子
2019/11/18 全球购物
行政管理专业推荐信
2013/11/02 职场文书
应届本科生推荐信范文
2013/12/25 职场文书
拓展训练激励口号
2014/06/17 职场文书
教师见习报告范文
2014/11/03 职场文书
护林员个人总结
2015/03/04 职场文书
2019单位介绍信怎么写
2019/06/24 职场文书
浅谈如何保证Mysql主从一致
2022/03/13 MySQL
星际争霸:毕姥爷vs解冻01
2022/04/01 星际争霸
Redis 限流器
2022/05/15 Redis