android - Tensorflow - 如何卡住 SavedModel 中的 .pb 以用于 TensorFlowInferenceInterface 中的推理？

Question

根据 this answer , 我可以提取 MetaGraph来自SavedModel , 然后卡住 MetaGraph的 GraphDef ，然后运行 ​​freeze_graph.py上面的脚本GraphDef得到.pb可在安卓系统中使用。我的问题:我究竟该如何提取 MetaGraph (然后是 GraphDef )？因为tf.saved_model.loader.load(sess, [tag_constants.SERVING], <model_path>)返回 MetaGraphDef而不是 MetaGraph .

Answer 1

刚收到。事实证明，在删除我从 conda 获得的 Tensorflow 版本并将其替换为来自 pip 的版本后，我可以这样做:

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 的那个不完整，即使安装了 pip，我也不得不复制 freeze_graph.py 和saved_model_utils 来自 tensorflow-master。此外，上面的代码大部分是从 freeze_graph_test.py 复制而来。