tensorflow读取inception_v1.ckpt

inception_v1.ckpt的下载地址
http://download.tensorflow.org/models/inception_v1_2016_08_28.tar.gz
并存入解压，存入文件夹中
我存储在’./tmp/checkpoints’

1. 导入包

from tensorflow.python import pywrap_tensorflow
import os

2. 得到文件夹的绝对路径

#相对路径
checkpoints_dir = './tmp/checkpoints'
#绝对路径
checkpoint_path = os.path.join(os.path.abspath(checkpoints_dir), "inception_v1.ckpt")

3. 获取一个reader，并获取权重的字典

reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
#返回的是一个字典
var_to_shape_map = reader.get_variable_to_shape_map()

4. 根据上一步返回字典获取节点name和value

for key in var_to_shape_map:
    print("tensor_name: ", key)
    print(reader.get_tensor(key))

get the name and value of op

5. 根据上一步返回字典获取节点name和value

for key,values in var_to_shape_map.items():
    try:
        print('/---------------/')
        print(key)
        print(values)
    except:
        pass

get the name and shape of op

全部代码：

import tensorflow as tf
from tensorflow.python import pywrap_tensorflow
import os
checkpoints_dir = './tmp/checkpoints'
checkpoint_path = os.path.join(os.path.abspath(checkpoints_dir), "inception_v1.ckpt")
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()#返回的是一个字典
for key in var_to_shape_map:
    print("tensor_name: ", key)
    print(reader.get_tensor(key)) # Remove this is you want to print only variable names
for key,values in var_to_shape_map.items():
    try:
        print('/---------------/')
        print(key)
        print(values)
    except:
        pass