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:?
??????????