tensorflow-serving - session_bundle 中的 manifest_pb2 在哪里？

Question

我正在尝试导出之前训练过的模型 (pb) 以用于服务使用以下代码片段

from tensorflow_serving.session_bundle import exporter
def create_graph():
        """Creates a graph from saved GraphDef file and returns a saver."""
        # Creates graph from saved graph_def.pb.
        with tf.gfile.FastGFile(modelFullPath, 'rb') as f:
            graph_def = tf.GraphDef()
            graph_def.ParseFromString(f.read())
            _ = tf.import_graph_def(graph_def, name='')

def export():
        print 'Exporting trained model to', export_path
        saver = tf.train.Saver(sharded=True)
        model_exporter = exporter.Exporter(saver)
        signature = exporter.classification_signature(input_tensor=x, scores_tensor=y)
        model_exporter.init(sess.graph.as_graph_def(),default_graph_signature=signature) 
        model_exporter.export(export_path, tf.constant(FLAGS.export_version), sess)
        print 'Done exporting!'

但是找不到 exporter.py 中的 manifest_pb2。我是否遗漏了这种方法中的一些基本内容？

Answer 1

manifest_pb2.py 是在使用 bazel 构建 session_bundle 时从 manifest.proto 生成的。在 Mac 上，我使用 pip3 安装了 TensorFlow，文件位于 /usr/local/lib/python3.5/site-packages/tensorflow/contrib/session_bundle/manifest_pb2.py。

按照 https://www.tensorflow.org/versions/r0.10/get_started/os_setup.html 中的步骤操作适用于您的平台，它应该可以解决问题。