对保存的参数checkpoints进行可视化读取 1.pywrap_tensorflow.NewCheckpoint(获得checkpoint的读取器) 2.np.save(对npy文件进行保存) 3.tl.file.load_npy_to

1. pywrap_tensorflow.NewCheckpoint(path)获得checkpoint的读取器

参数说明： path表示checkpoint的路径

2.np.save(path, dict) 根据路径将数据保存为npy类型

参数说明：path表示进行参数保存的路径, dict 表示需要进行保存的参数

3.tl.file.load_npy_to_any(name=path)对保存的npy文件进行读取

参数说明：name=path表示进行参数读取的路径

代码说明：

第一步：使用pywrap_tensorflow.NewCheckpoint(path)获得checkpoint的参数读取器

第二步：使用reader.get_variable_to_shape_map()构造字典

第三步：循环key，将键值对写入到all_variable.npy

第四步：使用tl.file.load_npy_to_any将npy数据进行读取

from tensorflow.python import pywrap_tensorflow

import os

import numpy as np

import tensorlayer as tl

#print出ckpt里的所有变量

# 第一步：构建读取checkpoint的reader

model_dir = './models'

checkpoints = model_dir + os.path.sep + 'model-20180626-205832.ckpt-60000'

reader = pywrap_tensorflow.NewCheckpointReader(checkpoints)
# 第二步：构建参数字典

var_to_shape_map = reader.get_variable_to_shape_map() # 存储所有变量

# 第三步：循环key，构建数据，使用np.save()进行数据保存

for key in var_to_shape_map:

    var_to_shape_map[key] = reader.get_tensor(key)

    np.save('all_variable.npy', var_to_shape_map)

# 第四步:使用tl.files.load_npy_to_any进行数据的读取

# data2 = np.load('./all_variable.npy', allow_pickle=True)

data = tl.files.load_npy_to_any(name='all_variable.npy')

for key, value in data.items():

    print(key, value.shape)

巴特西

对保存的参数checkpoints进行可视化读取 1.pywrap_tensorflow.NewCheckpoint(获得checkpoint的读取器) 2.np.save(对npy文件进行保存) 3.tl.file.load_npy_to_any(对保存的npy文件进行读取)

最新文章

热门文章