Tensorflow1.x 和 2.x如何读取ckpt中保存了那些参数

对于1.x的tf版本,网上已经有很多介绍了,这里给出代码:

import tensorflow as tf  import os
from tensorflow.python import pywrap_tensorflowmodel_dir='./model'#设置模型所在文件夹
checkpoint_path = os.path.join(model_dir, "fineturing_model.ckpt")#定位ckpt文件
# 从checkpoint中读出数据
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
# reader = tf.train.NewCheckpointReader(checkpoint_path) # 用tf.train中的NewCheckpointReader方法
var_to_shape_map = reader.get_variable_to_shape_map()
# 输出权重tensor名字和值
for key in var_to_shape_map:print("tensor_name: ", key,reader.get_tensor(key).shape)

对于tensorflow2.0来说,这个代码是不能正常运行的,会报类似:

module 'tensorflow.python.pywrap_tensorflow' has no attribute 'NewCheckpointReader'

的错误,这就是版本的问题,有些博主建议降低tf版本,但是这有点代价太大,下面给出tf2.0的正确使用方式;

import tensorflow.compat.v1 as tf1tf1.disable_v2_behavior()
checkpoint_path = 'xxx.ckpt'
# Read data from checkpoint file
reader = tf1.train.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:print("tensor_name: ", key)print(reader.get_tensor(key).shape)

Tensorflow1.x 和 2.x如何读取ckpt中保存了那些参数相关推荐

python文件读取方法read(size)的含义是_在Python中可使用read([size])来读取文件中的数据,如果参数size省略,则读取文件中的()。...
[单选题]李明在他所属的公司工作五年,每天都很认真地处理繁杂的事情,同事们都夸他认真,但是依然没有建树,这是因为: [多选题]品牌标志的作用表现在 [单选题]新产品开发的第一个阶段是_______. ...
python文件读取方法read(size)的含义是_在Python中可使用read([size])来读取文件中的数据,如果参数size省略,则读取文件中的()。(4.0分)_学小易找答案...
[单选题]文本文件存储的是(),由若干文本行组成,通常每行以换行符 '\n' 结尾.(4.0分) [单选题]()属性是返回被打开文件的访问模式.(4.0分) [单选题]重力坝是由砼或( )修筑而成的大 ...
matlab 保存三维矩阵,如何以Python语法读取Matlab中保存的三维矩阵（Dicom矩阵）？...
我在Matlab中保存了一个坐标(行=288,列=288,切片(z)=266)的3D矩阵. 现在我想用Python加载它.不幸的是,加载后,在Python中它是(row=288,col=266,sli ...
@PropertiesSource注解读取配置文件中的数据
这是properties配置文件. 数据结构. 注入对象. 或者: 使用对象获取属性值. 或者: 了解:========================================= 通过@Pro ...
如何用python读取文本中指定行的内容
如何用python读取文本中指定行的内容 搜索资料 我来答分享新浪微博 QQ空间浏览 5284 次查看全文 http://www.taodudu.cc/news/show-64036.ht ...
android读取工程目录下的文件,Android编程实现读取工程中的txt文件功能
本文实例讲述了Android编程实现读取工程中的txt文件功能.分享给大家供大家参考,具体如下: 1. 众所周知,Android的res文件夹是用来存储资源的,可以在res文件夹下建立一个raw文件夹 ...
pandas使用read_csv读取文件数据、设置converters参数将百分比字符串转换为数字
pandas使用read_csv读取文件数据.设置converters参数将百分比字符串转换为数字目录 pandas使用read_csv读取文件数据.设置converters参数将百分比字符串转换为 ...
SpringBoot 读取配置文件中参数全面教程
一.简介在日常开发使用 SpringBoot 框架时,经常有一些配置信息需要放置到配置文件中,我们需要手动读取这些配置到应用中进行一些逻辑,这里整理了一些常用读取配置的方法,简单介绍一下. 1.Sp ...
读取Assets中的文件数据
首先在assets文件夹中新建一个text.text文件用来读取读取文件代码如下: import java.io.BufferedReader; import java.io.IOException ...

Tensorflow1.x 和 2.x如何读取ckpt中保存了那些参数

Tensorflow1.x 和 2.x如何读取ckpt中保存了那些参数相关推荐

最新文章

热门文章