查看tensorflow中.ckpt文件的数据

最新推荐文章于 2024-10-28 21:11:32 发布

原创最新推荐文章于 2024-10-28 21:11:32 发布 · 5.6k 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow

深度学习专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一种使用TensorFlow从.ckpt文件中读取数据的方法。通过NewCheckpointReader类实例化对象，可以获取变量名和形状映射，并进一步提取、展平数据。文中展示了如何打印这些数据及其统计信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#获取.ckpt文件里面的数据
import os
import path
import numpy as np
from tensorflow.python import pywrap_tensorflow
checkpoint_path='/home/hisignal/python_work/my_ob_detect/model/ssd_mobilenet_v1_coco_2018_01_28/model.ckpt'
# print(path.getcwdu())
print(checkpoint_path)
#read data from checkpoint file
reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map=reader.get_variable_to_shape_map()
 
 
data_print=np.array([])
for key in var_to_shape_map:
    print('tensor_name',key)
    ckpt_data=np.array(reader.get_tensor(key))#cast list to np arrary
    ckpt_data=ckpt_data.flatten()#flatten list
    data_print=np.append(data_print,ckpt_data,axis=0)
 
 
print(data_print,data_print.shape,np.max(data_print),np.min(data_print),np.mean(data_print))