【keras】打印h5模型的保存下的变量名称及变量值
有时候需要直接打开h5文件观察权重,经过琢磨,发现可以如下打印:
import h5py
f = h5py.File('resnet50_weights_tf_dim_ordering_tf_kernels.h5','r') #打开h5文件
for key in f.keys():
for subkey in f[key]:
print(subkey )
print(f[key][subkey].value)
print(f[key][subkey].shape)
==========================
如果出现报错,关于group类型不存在value,shape等错误,
可自行加一层循环去读取类似字典的group