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使用点操作符访问字典(dict)数据的方法
Mar 16 Python
用Python写王者荣耀刷金币脚本
Dec 21 Python
python之pexpect实现自动交互的例子
Jul 25 Python
python图像处理模块Pillow的学习详解
Oct 09 Python
解决pandas展示数据输出时列名不能对齐的问题
Nov 18 Python
基于Python实现扑克牌面试题
Dec 11 Python
Python打印特殊符号及对应编码解析
May 07 Python
将tf.batch_matmul替换成tf.matmul的实现
Jun 18 Python
python如何获得list或numpy数组中最大元素对应的索引
Nov 16 Python
python3实现简单飞机大战
Nov 29 Python
实操Python爬取觅知网素材图片示例
Nov 27 Python
python创建字典及相关管理操作
Apr 13 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文件缓存内容保存格式实例分析
2014/08/20 PHP
如何使用Gitblog和Markdown建自己的博客
2015/07/31 PHP
PHP中__autoload和Smarty冲突的简单解决方法
2016/04/08 PHP
作为PHP程序员你要知道的另外一种日志
2018/07/30 PHP
Laravel5.5以下版本中如何自定义日志行为详解
2018/08/01 PHP
Laravel如何同时连接多个数据库详解
2019/08/13 PHP
JavaScript 基础问答三
2008/12/03 Javascript
一个可以随意添加多个序列的tag函数
2009/07/21 Javascript
javascript实现当前页导航激活的方法
2015/02/27 Javascript
jQuery插件animateSlide制作多点滑动幻灯片
2015/06/11 Javascript
讲解vue-router之什么是动态路由
2018/05/28 Javascript
js实现图片放大并跟随鼠标移动特效
2019/01/18 Javascript
react同构实践之实现自己的同构模板
2019/03/13 Javascript
layui的布局和表格的渲染以及动态生成表格的方法
2019/09/18 Javascript
Vue开发中遇到的跨域问题及解决方法
2020/02/11 Javascript
OpenLayers3实现鼠标移动显示坐标
2020/09/25 Javascript
Python 转义字符详细介绍
2017/03/21 Python
Python排序算法之选择排序定义与用法示例
2018/04/29 Python
python bmp转换为jpg 并删除原图的方法
2018/10/25 Python
python中的itertools的使用详解
2020/01/13 Python
python tkinter之 复选、文本、下拉的实现
2020/03/04 Python
python datetime处理时间小结
2020/04/16 Python
html5 canvas实现圆形时钟代码分享
2013/12/25 HTML / CSS
使用postMessage让 iframe自适应高度的方法示例
2019/10/08 HTML / CSS
英国高档时尚男装购物网站:MR PORTER
2016/08/09 全球购物
日本最大级玩偶手办购物:あみあみ Amiami
2018/04/23 全球购物
党员年终民主评议的自我评价
2013/11/05 职场文书
博士学位自我鉴定范文
2013/12/26 职场文书
超市端午节活动方案
2014/01/23 职场文书
小学三好学生事迹材料
2014/08/15 职场文书
党员查摆剖析材料
2014/10/10 职场文书
2019最新版劳务派遣管理制度
2019/08/16 职场文书
Python 把两层列表展开平铺成一层(5种实现方式)
2021/04/07 Python
使用pycharm运行flask应用程序的详细教程
2021/06/07 Python
python百行代码实现汉服圈图片爬取
2021/11/23 Python
MySQL日期时间函数知识汇总
2022/03/17 MySQL