tensorflow-ckpt2npy

最新推荐文章于 2020-12-02 02:45:42 发布

WeiGuanQun

最新推荐文章于 2020-12-02 02:45:42 发布

阅读量3.3k

点赞数

分类专栏： tensorflow 文章标签： python numpy tensorflow ckpt npy

本文链接：https://blog.csdn.net/weiguanqun/article/details/78673620

版权

tensorflow 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

这是一个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)