tensorflow 从ckpt中获取权重

仅从ckpt中获取权重名称/形状/tensor

import tensorflow as tf

from tensorflow.python import pywrap_tensorflow

reader=pywrap_tensorflow.NewCheckpointReader(r'/home/*/*/model/resnet_v2_101_2017_04_14/resnet_v2_101.ckpt')

all_var=reader.get_variable_to_shape_map()

for key in all_var:  # same as "for key in all_var.keys():"#

    tensor=reader.get_tensor(key)

    print(tensor)

or

reader=tf.train.NewCheckpointReader('*.ckpt')

all_var=reader.get_variable_to_shape_map()

tensor=reader.get_tensor(key)