使用现有C ++程序(不是Bazel)使用TensorFlow训练模型进行预测

时间:2016-10-13 22:56:48

标签: c++ tensorflow

我有一个使用Automake构建的大型C ++程序,将它转换为使用Bazel构建系统将是一个非常大的麻烦(在我的时间限制下几乎不可能)。有没有什么方法可以在我的程序中使用TensorFlow训练模型(深度卷积网)来进行预测(我现在不需要在程序中学习,但如果也可以这样做真的很酷) ?比如使用TensorFlow作为库?

谢谢!

1 个答案:

答案 0 :(得分:0)

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