Tensorflow - TensorFlowInferenceInterfaceの推論に使用するSavedModelから.pbをフリーズする方法

Question

this answerによると、私はMetaGraphさんGraphDefを凍結し、その後、アンドロイドに.pbが使用可能取得することGraphDefにfreeze_graph.pyスクリプトを実行し、その後、SavedModelからMetaGraphを抽出することができます。私の質問：MetaGraph（正確にはGraphDef）をどのように抽出すればよいですか？ tf.saved_model.loader.load(sess, [tag_constants.SERVING], <model_path>)はMetaGraphの代わりにMetaGraphDefを返します。

Answer 1

私はちょうどそれを手に入れました。私はcondaからもらったTensorflowのバージョンを削除し、pipから1と交換した後、私はこれを行うことが、判明：

from tensorflow.python.tools import freeze_graph 
from tensorflow.python.saved_model import tag_constants 

input_saved_model_dir = "F:/python_machine_learning_codes/estimator_exported_model/1509418513" 
output_node_names = "softmax_tensor" 
input_binary = False 
input_saver_def_path = False 
restore_op_name = None 
filename_tensor_name = None 
clear_devices = False 
input_meta_graph = False 
checkpoint_path = None 
input_graph_filename = None 
saved_model_tags = tag_constants.SERVING 

freeze_graph.freeze_graph(input_graph_filename, input_saver_def_path, 
          input_binary, checkpoint_path, output_node_names, 
           restore_op_name, filename_tensor_name, 
           output_graph_filename, clear_devices, "", "", "", 
           input_meta_graph, input_saved_model_dir, 
          saved_model_tags) 

conda-forgeから1が不完全だった、とさえpipでインストールし、私が持っていましたfreeze_graph.pyとsaved_model_utilsをtensorflow-masterからコピーします。また、上のコードはほとんどがfreeze_graph_test.pyからコピーされています。