TensorFlow把ckpt转化为pb文件代码进阶版

把tensorflow保存的checkpoint类型模型冻结，转化为.pb模型输出。调用方法参见我的博文https://blog.csdn.net/weixin_41864878/article/details/84957681

import argparse
import tensorflow as tf
from tensorflow.python.framework import graph_util

def freeze_graph（input_checkpointoutput_graph output_node_names）:
    ‘‘‘
    To convert the TensorFlow checkpoint files to froze graph
    :param input_checkpoint:
    :param output_graph:
    :return:
    ‘‘‘
    saver = tf.train.import_meta_graph（input_checkpoint + ‘.meta‘ clear_devices=True）
 
    with tf.Session（） as sess:
        saver.restore（sess input_checkpoint）
        output_graph_def = graph_util.convert_variables_to_constants（  
            sess=sess
            input_graph_def=sess.graph_def# :sess.graph_def
            output_node_names=output_node_names.split（““））#
 
        with tf.gfile.GFile（output_graph “wb“） as f: