TensorFlow中查看checkpoint文件中的变量名和对应值

在加载模型时，需要知道checkpoint中变量名称，一下代码可以查看TensorFlow中checkpoint文件中的变量名：

#!/usr/bin/env python
# -*- coding:utf-8 -*-
import os
from tensorflow.python import pywrap_tensorflow
model_dir = "Saved_model"
checkpoint_path = os.path.join(model_dir, "model.ckpt")
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:print("tensor_name: ", key, end=' ')print(reader.get_tensor(key))

代码说明：

在上述代码执行文件中，有一个保存的路径为“Saved_model/model.ckpt”的模型，具体信息如下图suo

你可以根据你自己的路径来修改其中的参数即可查看你自己的checkpoint文件中的变量名和变量值。

TensorFlow中查看checkpoint文件中的变量名和对应值相关推荐

python db文件_python中查看.db文件中表格的名字及表格中的字段操作
1.问题描述: 我桌面上有一个"账号密码.db"文件,我现在想知道里面有几张表格table.表格的名字.表头结构. 2.使用SQL语句"""selec ...
android怎么查看方法被谁调用,Android中查看布局文件中的控件（view,id）在哪里被调用（使用）...
在阅读别人的代码时通常是很痛苦的,有时很想要看一看布局中的控件在哪里被调用了,为之很苦恼在这里提供一种方法. 复制要查看的控件ID,到R文件中搜索到该ID, 接下来就好办的了,选中ID按下Ctrl鼠 ...
html img调用js,html调用js变量如何在html中输出js文件中的变量
html页面代码中怎么调用js变量?html页面代码中怎么调用js变量,例如在html代码中插入js代码: a=取浏览你把index1.js 中的onReady 去掉,把index1.js改成 fu ...
【Ubuntu】dpkg-deb -c :查看deb文件中的内容
1.dpkg-deb -c :查看deb文件中的内容 $ dpkg-deb -c package eg: $ dpkg-deb -c sogoupinyin_2.2.0.0108_amd64.deb ...
如何在Web浏览器中查看XML文件
一旦了解到HTML是XML的一种形式,您可能会想知道如果尝试在浏览器中查看XML文件会发生什么. 结果令人非常失望-Firefox在页面顶部显示一条横幅,上面写着:"此XML文件似乎没有与之 ...
在Qt中查看.raw文件
在Qt中查看.raw文件 Qt中图像类为QImage,而OpenCV对图像操作使用Mat/IplImage,因此,想要在Qt的控件上查看.raw文件需要先将IplImage/Mat转换为QImage ...
html中引入txt文件,如何导入放置在txt文件中的.html文件中的代码？
所以,我想要的是这个.我有一个HTML文件,我想从代码加载文本文件.因此,浏览器应该从该文本文件中获取代码并将其作为HTML代码的一部分进行阅读.这里有一个例子:如何导入放置在txt文件中的.html ...
vscode中如何创新建php文件,php – 如何在Visual Studio Code,UNIX中的所有文件中创建所有行结尾(EOL)？...
我使用Windows 10 home,我通常使用Visual Studio Code(VSCODE)来编辑Linux Bash脚本以及PHP和JavaScript. 我没有开发任何专门用于Window ...
Java中的properties文件中的key不能使用项目中的接口名和Java文件名
Java中的properties文件中的key不能使用项目中的接口名和Java文件名,否则在反射生成实体类的时候会出现报错的情况.

TensorFlow中查看checkpoint文件中的变量名和对应值

TensorFlow中查看checkpoint文件中的变量名和对应值相关推荐

最新文章

热门文章