TensorFlow ckpt模型文件转换为pb模型文件

问题

ckpt转pb主要有两点要注意

1.要知道模型输入输出的节点名称

使用tools里的freeze_graph来读取ckpt，展示所有节点名称，查找搜需要的

2.转化后Value Error问题

Value Error错误信息一般为
ValueError: Input 0 of node … was passed float from … incompatible with expected float_ref.
对其进行类型转换即可

1.获取输入输出节点名称

使用tools里的freeze_graph来读取ckpt

保存ckpt模型的文件夹下的三个文件名
epoch_50.ckpt.data-00000-of-00001
epoch_50.ckpt.index
epoch_50.ckpt.meta

获取所有节点名称

from tensorflow.python.tools import freeze_graph

def freeze_graph_name(input_checkpoint):
    '''
    :param input_checkpoint: 
    '''
    saver = tf.train.import_meta_graph(input_checkpoint + '.meta', clear_devices=True)
    graph = tf.get_default_graph()
    input_graph_def = graph.as_graph_def()
    with tf.Session() as sess:
        saver.rest