tensorflow-ckpt2npy

这是一个ckpt转npy的小程序

from tensorflow.python import pywrap_tensorflow
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)