将重新训练的数据转换为.pb格式的ios相机示例？

Question

我使用教程（（https://www.tensorflow.org/versions/r0.9/how_tos/image_retraining/index.html）重新训练了图像数据）我做了所有的步骤直到

  

bazel build tensorflow / examples / image_retraining：retrain

。我想知道如何将此图像训练数据转换为我可以在ios相机示例中使用的.pb文件。

感谢您的帮助！

Answer 1

我认为这是一个两步过程。

1. 导出模型定义和权重：

   一个。使用tf.train.write_graph的graphdef（* .pb）：https://www.tensorflow.org/versions/r0.11/api_docs/cc/index.html

   湾权重（* .ckpt）使用tf.train.Saver：https://www.tensorflow.org/versions/r0.11/api_docs/python/state_ops.html#Saver

2. 使用以下脚本将上述两个组件合并为一个protobuf（final * .pb）：https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py