【发布时间】:2016-10-15 07:08:03
【问题描述】:
我有一个用 Automake 构建的大型 C++ 程序,将它转换为使用 Bazel 构建系统将是一个非常大的麻烦(考虑到我的时间限制,实际上几乎不可能)。有什么方法可以在我的程序中使用 TensorFlow 训练的模型(深度卷积网络)进行预测(我现在不需要在程序中进行学习,但如果也可以这样做,那就太酷了) ?喜欢使用 TensorFlow 作为库吗?
谢谢!
【问题讨论】:
标签: c++ tensorflow
我有一个用 Automake 构建的大型 C++ 程序,将它转换为使用 Bazel 构建系统将是一个非常大的麻烦(考虑到我的时间限制,实际上几乎不可能)。有什么方法可以在我的程序中使用 TensorFlow 训练的模型(深度卷积网络)进行预测(我现在不需要在程序中进行学习,但如果也可以这样做,那就太酷了) ?喜欢使用 TensorFlow 作为库吗?
谢谢!
【问题讨论】:
标签: c++ tensorflow
TensorFlow 在 c_api.h 头文件中导出了一个非常狭窄的 C API。可以通过构建libtensorflow.so 目标来生成库文件。 C API 拥有加载预训练模型并在其上运行推理以进行预测所需的一切。您不需要将您的构建系统转换为 Bazel。您只需使用 Bazel 构建 //tensorflow:libtensorflow.so 目标,并将 libtensorflow.so 和 c_api.h 复制到您认为合适的位置。
【讨论】: