这是一个ckpt转npy的小程序
import numpy as np
checkpoint_path='/home/wgq/PycharmProjects/tensorflow_cpkt2npy/model/front/model_epoch100.ckpt'# your ckpt path
reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map=reader.get_variable_to_shape_map()
vgg19={}
for key in var_to_shape_map:
print ("tensor_name",key)
sStr_2=key[:-2]
print sStr_2
if not vgg19.has_key(sStr_2):
vgg19[sStr_2]=[reader.get_tensor(key)]
else:
vgg19[sStr_2].append(reader.get_tensor(key))
np.save(' name.npy',vgg19)