Tensorflow 将MetaGraphDef转换为GraphDef的简单方法?

Tensorflow 将MetaGraphDef转换为GraphDef的简单方法?,tensorflow,Tensorflow,有没有一种简单的方法可以将SaveModel生成的MetaGraphDef原型传输到GraphDef tf.saved_model.builder.SavedModelBuilder我在这里找到了答案: 所以基本上 graph_def = meta_graph_def.graph_def 你可以推断出这个答案 metagraphdef-->graphdef-->优化-->metagraphdef from tensorflow.python.tools import freeze_grap

有没有一种简单的方法可以将SaveModel生成的MetaGraphDef原型传输到GraphDef


tf.saved_model.builder.SavedModelBuilder

我在这里找到了答案:

所以基本上

graph_def = meta_graph_def.graph_def

你可以推断出这个答案

metagraphdef-->graphdef-->优化-->metagraphdef

from tensorflow.python.tools import freeze_graph
def freeze_model(saved_model_dir, output_node_names, output_filename):
  output_graph_filename = os.path.join(saved_model_dir, output_filename)
  initializer_nodes = ''
  freeze_graph.freeze_graph(
      input_saved_model_dir=saved_model_dir,
      output_graph=output_graph_filename,
      saved_model_tags = tag_constants.SERVING,
      output_node_names=output_node_names,
      initializer_nodes=initializer_nodes,
      input_graph=None,
      input_saver=False,
      input_binary=False,
      input_checkpoint=None,
      restore_op_name=None,
      filename_tensor_name=None,
      clear_devices=False,
      input_meta_graph=False,
  )
  print('graph freezed!')

freeze_graph(saved_model_dir, “head/predictions/class_ids”)

欢迎链接到答案和信息。但是请添加一些上下文来支持它,以防链接断开。