如何在 C++ 中使用 TensorFlow Estimator?
Posted
技术标签:
【中文标题】如何在 C++ 中使用 TensorFlow Estimator?【英文标题】:How to use TensorFlow Estimator from C++? 【发布时间】:2018-06-24 16:39:22 【问题描述】:我对 TensorFlow 很陌生,但我必须用 C++ 实现一个 TensorFlow 模型。
我假设我可以用 Python 定义一个模型(图)并将其导出,然后按照本文所述从 C++ 调用导出的图。
Loading a TensorFlow graph with the C++ API
但是——由于我对它的底层理解有限——我不确定如何使用Estimator 做同样的事情。这是因为显然我没有明确的“会话”和“图表”定义(不过我知道它被包裹在里面)。
您能在这里指导一下吗?
【问题讨论】:
【参考方案1】:由于C++没有高级api(reference),而且文中使用的tf.Session比Estimator的低层,我个人认为这是不可能的。
我最初的假设是将 TensorFlow Serving 与 C++ 客户端 (example) 一起使用,但任何想法都非常受欢迎。
【讨论】:
以上是关于如何在 C++ 中使用 TensorFlow Estimator?的主要内容,如果未能解决你的问题,请参考以下文章
如何在 C++ 中保存和恢复 TensorFlow 图及其状态?
如何构建和使用 Google TensorFlow C++ api
如何在 C++ 代码中运行 Yolov5 tensorflow model.pb?