TensorFlow学习笔记：获取inception模型全部te

tf.Graph.get_tensor_by_name()

All tensors have string names which you can see as follows:

[tensor.name for tensor in tf.get_default_graph().as_graph_def().node]

Once you know the name you can fetch the Tensor using <name>:0 (0 refers to endpoint which is somewhat redundant)

import tensorflow as tf

c = tf.constant([[1.0, 2.0], [3.0, 4.0]])
d = tf.constant([[1.0, 1.0], [0.0, 1.0]])
e = tf.matmul(c, d, name='example')

with tf.Session() as sess:
    test = sess.run(e)
    print (e.name)  
    # example:0
    # <name>:0 (0 refers to endpoint which is somewhat redundant)
    # 形如'conv1'是节点名称，而'conv1:0'是张量名称，表示节点的第一个输出张量
    test = tf.get_default_graph().get_tensor_by_name("example:0")
    print (test)    
    # Tensor("example:0", shape=(2, 2), dtype=float32)

实例：

以inception retrain为例，列出修改后的inception模型全部tensor：

# -*- coding: utf-8 -*-
"""
Created on Fri Oct 13 20:38:15 2017
@author: Dexter
"""
import tensorflow as tf
import os

model_dir = 'C:/Users/Dexter/Documents/ML_files/171009_Retrain_with_Inception_v3'
model_name = 'output_graph.pb'

# 读取并创建一个图graph来存放Google训练好的Inception_v3模型（函数）
def create_graph():
    with tf.gfile.FastGFile(os.path.join(
            model_dir, model_name), 'rb') as f:
        # 使用tf.GraphDef()定义一个空的Graph
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        # Imports the graph from graph_def into the current default Graph.
        tf.import_graph_def(graph_def, name='')

# 创建graph
create_graph()

tensor_name_list = [tensor.name for tensor in tf.get_default_graph().as_graph_def().node]
for tensor_name in tensor_name_list:
    print(tensor_name,'\n')

输出：

mixed_8/tower/conv_1/batchnorm/moving_variance 

mixed_8/tower/conv_1/batchnorm 

mixed_8/tower/conv_1/CheckNumerics 

mixed_8/tower/conv_1/control_dependency 

mixed_8/tower/conv_1 

mixed_8/tower_1/conv/conv2d_params 

mixed_8/tower_1/conv/Conv2D 

mixed_8/tower_1/conv/batchnorm/beta 

mixed_8/tower_1/conv/batchnorm/gamma 

mixed_8/tower_1/conv/batchnorm/moving_mean 

mixed_8/tower_1/conv/batchnorm/moving_variance 

mixed_8/tower_1/conv/batchnorm 

mixed_8/tower_1/conv/CheckNumerics 

mixed_8/tower_1/conv/control_dependency 

mixed_8/tower_1/conv 

mixed_8/tower_1/conv_1/conv2d_params 

mixed_8/tower_1/conv_1/Conv2D 

mixed_8/tower_1/conv_1/batchnorm/beta 

mixed_8/tower_1/conv_1/batchnorm/gamma 

mixed_8/tower_1/conv_1/batchnorm/moving_mean 

mixed_8/tower_1/conv_1/batchnorm/moving_variance 

mixed_8/tower_1/conv_1/batchnorm 

mixed_8/tower_1/conv_1/CheckNumerics 

mixed_8/tower_1/conv_1/control_dependency 

mixed_8/tower_1/conv_1 

mixed_8/tower_1/conv_2/conv2d_params 

mixed_8/tower_1/conv_2/Conv2D 

mixed_8/tower_1/conv_2/batchnorm/beta 

mixed_8/tower_1/conv_2/batchnorm/gamma 

mixed_8/tower_1/conv_2/batchnorm/moving_mean 

mixed_8/tower_1/conv_2/batchnorm/moving_variance 

mixed_8/tower_1/conv_2/batchnorm 

mixed_8/tower_1/conv_2/CheckNumerics 

mixed_8/tower_1/conv_2/control_dependency 

mixed_8/tower_1/conv_2 

mixed_8/tower_1/conv_3/conv2d_params 

mixed_8/tower_1/conv_3/Conv2D 

mixed_8/tower_1/conv_3/batchnorm/beta 

mixed_8/tower_1/conv_3/batchnorm/gamma 

mixed_8/tower_1/conv_3/batchnorm/moving_mean 

mixed_8/tower_1/conv_3/batchnorm/moving_variance 

mixed_8/tower_1/conv_3/batchnorm 

mixed_8/tower_1/conv_3/CheckNumerics 

mixed_8/tower_1/conv_3/control_dependency 

mixed_8/tower_1/conv_3 

mixed_8/pool/CheckNumerics 

mixed_8/pool/control_dependency 

mixed_8/pool 

mixed_8/join/concat_dim 

mixed_8/join 

mixed_9/conv/conv2d_params 

mixed_9/conv/Conv2D 

mixed_9/conv/batchnorm/beta 

mixed_9/conv/batchnorm/gamma 

mixed_9/conv/batchnorm/moving_mean 

mixed_9/conv/batchnorm/moving_variance 

mixed_9/conv/batchnorm 

mixed_9/conv/CheckNumerics 

mixed_9/conv/control_dependency 

mixed_9/conv 

mixed_9/tower/conv/conv2d_params 

mixed_9/tower/conv/Conv2D 

mixed_9/tower/conv/batchnorm/beta 

mixed_9/tower/conv/batchnorm/gamma 

mixed_9/tower/conv/batchnorm/moving_mean 

mixed_9/tower/conv/batchnorm/moving_variance 

mixed_9/tower/conv/batchnorm 

mixed_9/tower/conv/CheckNumerics 

mixed_9/tower/conv/control_dependency 

mixed_9/tower/conv 

mixed_9/tower/mixed/conv/conv2d_params 

mixed_9/tower/mixed/conv/Conv2D 

mixed_9/tower/mixed/conv/batchnorm/beta 

mixed_9/tower/mixed/conv/batchnorm/gamma 

mixed_9/tower/mixed/conv/batchnorm/moving_mean 

mixed_9/tower/mixed/conv/batchnorm/moving_variance 

mixed_9/tower/mixed/conv/batchnorm 

mixed_9/tower/mixed/conv/CheckNumerics 

mixed_9/tower/mixed/conv/control_dependency 

mixed_9/tower/mixed/conv 

mixed_9/tower/mixed/conv_1/conv2d_params 

mixed_9/tower/mixed/conv_1/Conv2D 

mixed_9/tower/mixed/conv_1/batchnorm/beta 

mixed_9/tower/mixed/conv_1/batchnorm/gamma 

mixed_9/tower/mixed/conv_1/batchnorm/moving_mean 

mixed_9/tower/mixed/conv_1/batchnorm/moving_variance 

mixed_9/tower/mixed/conv_1/batchnorm 

mixed_9/tower/mixed/conv_1/CheckNumerics 

mixed_9/tower/mixed/conv_1/control_dependency 

mixed_9/tower/mixed/conv_1 

mixed_9/tower_1/conv/conv2d_params 

mixed_9/tower_1/conv/Conv2D 

mixed_9/tower_1/conv/batchnorm/beta 

mixed_9/tower_1/conv/batchnorm/gamma 

mixed_9/tower_1/conv/batchnorm/moving_mean 

mixed_9/tower_1/conv/batchnorm/moving_variance 

mixed_9/tower_1/conv/batchnorm 

mixed_9/tower_1/conv/CheckNumerics 

mixed_9/tower_1/conv/control_dependency 

mixed_9/tower_1/conv 

mixed_9/tower_1/conv_1/conv2d_params 

mixed_9/tower_1/conv_1/Conv2D 

mixed_9/tower_1/conv_1/batchnorm/beta 

mixed_9/tower_1/conv_1/batchnorm/gamma 

mixed_9/tower_1/conv_1/batchnorm/moving_mean 

mixed_9/tower_1/conv_1/batchnorm/moving_variance 

mixed_9/tower_1/conv_1/batchnorm 

mixed_9/tower_1/conv_1/CheckNumerics 

mixed_9/tower_1/conv_1/control_dependency 

mixed_9/tower_1/conv_1 

mixed_9/tower_1/mixed/conv/conv2d_params 

mixed_9/tower_1/mixed/conv/Conv2D 

mixed_9/tower_1/mixed/conv/batchnorm/beta 

mixed_9/tower_1/mixed/conv/batchnorm/gamma 

mixed_9/tower_1/mixed/conv/batchnorm/moving_mean 

mixed_9/tower_1/mixed/conv/batchnorm/moving_variance 

mixed_9/tower_1/mixed/conv/batchnorm 

mixed_9/tower_1/mixed/conv/CheckNumerics 

mixed_9/tower_1/mixed/conv/control_dependency 

mixed_9/tower_1/mixed/conv 

mixed_9/tower_1/mixed/conv_1/conv2d_params 

mixed_9/tower_1/mixed/conv_1/Conv2D 

mixed_9/tower_1/mixed/conv_1/batchnorm/beta 

mixed_9/tower_1/mixed/conv_1/batchnorm/gamma 

mixed_9/tower_1/mixed/conv_1/batchnorm/moving_mean 

mixed_9/tower_1/mixed/conv_1/batchnorm/moving_variance 

mixed_9/tower_1/mixed/conv_1/batchnorm 

mixed_9/tower_1/mixed/conv_1/CheckNumerics 

mixed_9/tower_1/mixed/conv_1/control_dependency 

mixed_9/tower_1/mixed/conv_1 

mixed_9/tower_2/pool 

mixed_9/tower_2/conv/conv2d_params 

mixed_9/tower_2/conv/Conv2D 

mixed_9/tower_2/conv/batchnorm/beta 

mixed_9/tower_2/conv/batchnorm/gamma 

mixed_9/tower_2/conv/batchnorm/moving_mean 

mixed_9/tower_2/conv/batchnorm/moving_variance 

mixed_9/tower_2/conv/batchnorm 

mixed_9/tower_2/conv/CheckNumerics 

mixed_9/tower_2/conv/control_dependency 

mixed_9/tower_2/conv 

mixed_9/join/concat_dim 

mixed_9/join 

mixed_10/conv/conv2d_params 

mixed_10/conv/Conv2D 

mixed_10/conv/batchnorm/beta 

mixed_10/conv/batchnorm/gamma 

mixed_10/conv/batchnorm/moving_mean 

mixed_10/conv/batchnorm/moving_variance 

mixed_10/conv/batchnorm 

mixed_10/conv/CheckNumerics 

mixed_10/conv/control_dependency 

mixed_10/conv 

mixed_10/tower/conv/conv2d_params 

mixed_10/tower/conv/Conv2D 

mixed_10/tower/conv/batchnorm/beta 

mixed_10/tower/conv/batchnorm/gamma 

mixed_10/tower/conv/batchnorm/moving_mean 

mixed_10/tower/conv/batchnorm/moving_variance 

mixed_10/tower/conv/batchnorm 

mixed_10/tower/conv/CheckNumerics 

mixed_10/tower/conv/control_dependency 

mixed_10/tower/conv 

mixed_10/tower/mixed/conv/conv2d_params 

mixed_10/tower/mixed/conv/Conv2D 

mixed_10/tower/mixed/conv/batchnorm/beta 

mixed_10/tower/mixed/conv/batchnorm/gamma 

mixed_10/tower/mixed/conv/batchnorm/moving_mean 

mixed_10/tower/mixed/conv/batchnorm/moving_variance 

mixed_10/tower/mixed/conv/batchnorm 

mixed_10/tower/mixed/conv/CheckNumerics 

mixed_10/tower/mixed/conv/control_dependency 

mixed_10/tower/mixed/conv 

mixed_10/tower/mixed/conv_1/conv2d_params 

mixed_10/tower/mixed/conv_1/Conv2D 

mixed_10/tower/mixed/conv_1/batchnorm/beta 

mixed_10/tower/mixed/conv_1/batchnorm/gamma 

mixed_10/tower/mixed/conv_1/batchnorm/moving_mean 

mixed_10/tower/mixed/conv_1/batchnorm/moving_variance 

mixed_10/tower/mixed/conv_1/batchnorm 

mixed_10/tower/mixed/conv_1/CheckNumerics 

mixed_10/tower/mixed/conv_1/control_dependency 

mixed_10/tower/mixed/conv_1 

mixed_10/tower_1/conv/conv2d_params 

mixed_10/tower_1/conv/Conv2D 

mixed_10/tower_1/conv/batchnorm/beta 

mixed_10/tower_1/conv/batchnorm/gamma 

mixed_10/tower_1/conv/batchnorm/moving_mean 

mixed_10/tower_1/conv/batchnorm/moving_variance 

mixed_10/tower_1/conv/batchnorm 

mixed_10/tower_1/conv/CheckNumerics 

mixed_10/tower_1/conv/control_dependency 

mixed_10/tower_1/conv 

mixed_10/tower_1/conv_1/conv2d_params 

mixed_10/tower_1/conv_1/Conv2D 

mixed_10/tower_1/conv_1/batchnorm/beta 

mixed_10/tower_1/conv_1/batchnorm/gamma 

mixed_10/tower_1/conv_1/batchnorm/moving_mean 

mixed_10/tower_1/conv_1/batchnorm/moving_variance 

mixed_10/tower_1/conv_1/batchnorm 

mixed_10/tower_1/conv_1/CheckNumerics 

mixed_10/tower_1/conv_1/control_dependency 

mixed_10/tower_1/conv_1 

mixed_10/tower_1/mixed/conv/conv2d_params 

mixed_10/tower_1/mixed/conv/Conv2D 

mixed_10/tower_1/mixed/conv/batchnorm/beta 

mixed_10/tower_1/mixed/conv/batchnorm/gamma 

mixed_10/tower_1/mixed/conv/batchnorm/moving_mean 

mixed_10/tower_1/mixed/conv/batchnorm/moving_variance 

mixed_10/tower_1/mixed/conv/batchnorm 

mixed_10/tower_1/mixed/conv/CheckNumerics 

mixed_10/tower_1/mixed/conv/control_dependency 

mixed_10/tower_1/mixed/conv 

mixed_10/tower_1/mixed/conv_1/conv2d_params 

mixed_10/tower_1/mixed/conv_1/Conv2D 

mixed_10/tower_1/mixed/conv_1/batchnorm/beta 

mixed_10/tower_1/mixed/conv_1/batchnorm/gamma 

mixed_10/tower_1/mixed/conv_1/batchnorm/moving_mean 

mixed_10/tower_1/mixed/conv_1/batchnorm/moving_variance 

mixed_10/tower_1/mixed/conv_1/batchnorm 

mixed_10/tower_1/mixed/conv_1/CheckNumerics 

mixed_10/tower_1/mixed/conv_1/control_dependency 

mixed_10/tower_1/mixed/conv_1 

mixed_10/tower_2/pool/CheckNumerics 

mixed_10/tower_2/pool/control_dependency 

mixed_10/tower_2/pool 

mixed_10/tower_2/conv/conv2d_params 

mixed_10/tower_2/conv/Conv2D 

mixed_10/tower_2/conv/batchnorm/beta 

mixed_10/tower_2/conv/batchnorm/gamma 

mixed_10/tower_2/conv/batchnorm/moving_mean 

mixed_10/tower_2/conv/batchnorm/moving_variance 

mixed_10/tower_2/conv/batchnorm 

mixed_10/tower_2/conv/CheckNumerics 

mixed_10/tower_2/conv/control_dependency 

mixed_10/tower_2/conv 

mixed_10/join/concat_dim 

mixed_10/join 

pool_3 

pool_3/_reshape/shape 

pool_3/_reshape 

input/BottleneckInputPlaceholder 

final_training_ops/weights/final_weights 

final_training_ops/weights/final_weights/read 

final_training_ops/biases/final_biases 

final_training_ops/biases/final_biases/read 

final_training_ops/Wx_plus_b/MatMul 

final_training_ops/Wx_plus_b/add 

final_result 

参考资料：

• Tensorflow: How to get a tensor by name?