【问题标题】:Making predictions with TensorFlow trained model with existing C++ program (not Bazel)使用现有 C++ 程序(不是 Bazel)使用 TensorFlow 训练模型进行预测
【发布时间】:2016-10-15 07:08:03
【问题描述】:

我有一个用 Automake 构建的大型 C++ 程序,将它转换为使用 Bazel 构建系统将是一个非常大的麻烦(考虑到我的时间限制,实际上几乎不可能)。有什么方法可以在我的程序中使用 TensorFlow 训练的模型(深度卷积网络)进行预测(我现在不需要在程序中进行学习,但如果也可以这样做,那就太酷了) ?喜欢使用 TensorFlow 作为库吗?

谢谢!

【问题讨论】:

    标签: c++ tensorflow


    【解决方案1】:

    TensorFlow 在 c_api.h 头文件中导出了一个非常狭窄的 C API。可以通过构建libtensorflow.so 目标来生成库文件。 C API 拥有加载预训练模型并在其上运行推理以进行预测所需的一切。您不需要将您的构建系统转换为 Bazel。您只需使用 Bazel 构建 //tensorflow:libtensorflow.so 目标,并将 libtensorflow.soc_api.h 复制到您认为合适的位置。

    【讨论】:

    • 非常感谢!我会试试的。
    • @kaveman 我可以再请求一个帮助吗?如果你能看看这个new question 并帮助我,我将不胜感激。
    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2020-04-18
    • 1970-01-01
    • 2020-07-04
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多