c++ - 如何使用 tensorflow c++ API 导入元图？

Question

我找到了一些关于如何在将模型导出到 .pb 文件后使用 Tensorflow c++ 导入模型的资源，但我的理解是 .pb file 方法已替换为更新的方法，该方法使用 tf.Saver.save 方法生成 .meta、.index、.data-00000-of-00001 和一个 checkpoint 文件。我找不到任何关于如何使用 C++ API 从这些文件类型导入模型的信息。

我该怎么做？

Answer 1

我在 Tensorflow 的顶部使用 TFLearn 包装器，但该过程应该与普通 Tensorflow 模型相同。您可以从 TFLearn in this way 中保存检查点正确，但你也可以 freeze如果你想使用 .pb 模型。可以同时加载 checkpoint或 model C++ 中的文件。无论如何，inference part在 C++ 中是相同的。