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通过pil模块将raw图片转换成png图片的方法
Mar 16 Python
python实现颜色空间转换程序(Tkinter)
Dec 31 Python
Python和Perl绘制中国北京跑步地图的方法
Mar 03 Python
Python升级导致yum、pip报错的解决方法
Sep 06 Python
Python基于递归实现电话号码映射功能示例
Apr 13 Python
详解python如何在django中为用户模型添加自定义权限
Oct 15 Python
浅谈pandas用groupby后对层级索引levels的处理方法
Nov 06 Python
python实现抖音点赞功能
Apr 07 Python
Python3 Tkinter选择路径功能的实现方法
Jun 14 Python
Python 把序列转换为元组的函数tuple方法
Jun 27 Python
Django框架模板用法入门教程
Nov 04 Python
Python数据分析入门之数据读取与存储
May 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
法国:浪漫之都的咖啡文化
2021/03/03 咖啡文化
php中通过Ajax如何实现异步文件上传的代码实例
2011/05/07 PHP
php学习笔记 php中面向对象三大特性之一[封装性]的应用
2011/06/13 PHP
PHP输出XML到页面的3种方法详解
2013/06/06 PHP
PHP 过滤页面中的BOM(实现代码)
2013/06/29 PHP
CodeIgniter开发实现支付宝接口调用的方法示例
2016/11/14 PHP
PHP中include()与require()的区别说明
2017/02/14 PHP
ExtJs3.0中Store添加 baseParams 的Bug
2010/03/10 Javascript
json的定义、标准格式及json字符串检验
2014/05/11 Javascript
node.js中的fs.symlinkSync方法使用说明
2014/12/15 Javascript
JavaScript、tab切换完整版(自动切换、鼠标移入停止、移开运行)
2016/01/05 Javascript
Jquery中巧用Ajax的beforeSend方法
2016/01/20 Javascript
Nodejs进阶:如何将图片转成datauri嵌入到网页中去实例
2016/11/21 NodeJs
Javascript设计模式之装饰者模式详解篇
2017/01/17 Javascript
JS实现复制功能
2017/03/01 Javascript
JS实现颜色的10进制转化成rgba格式的方法
2017/09/04 Javascript
vue项目中jsonp跨域获取qq音乐首页推荐问题
2018/05/30 Javascript
实例分析Array.from(arr)与[...arr]到底有何不同
2019/04/09 Javascript
详解vue更改头像功能实现
2019/04/28 Javascript
关于ligerui子页面关闭后,父页面刷新,重新加载的方法
2019/09/27 Javascript
vue中keep-alive内置组件缓存的实例代码
2020/04/16 Javascript
小程序实现图片移动缩放效果
2020/05/26 Javascript
[06:16]《DAC最前线》之地区预选赛全面回顾
2015/01/19 DOTA
python使用win32com在百度空间插入html元素示例
2014/02/20 Python
python从网络读取图片并直接进行处理的方法
2015/05/22 Python
python简单实例训练(21~30)
2017/11/15 Python
Django 连接sql server数据库的方法
2018/06/30 Python
python读取csv和txt数据转换成向量的实例
2019/02/12 Python
Matplotlib使用字符串代替变量绘制散点图的方法
2020/02/17 Python
Python如何生成xml文件
2020/06/04 Python
Html5在手机端调用相机的方法实现
2020/05/13 HTML / CSS
HTML5中外部浏览器唤起微信分享功能的代码
2020/09/15 HTML / CSS
红旗团支部事迹材料
2014/01/27 职场文书
《伯牙绝弦》教学反思
2014/03/02 职场文书
婚礼答谢宴主持词
2014/03/14 职场文书
个人安全生产承诺书
2014/05/22 职场文书