c++ - 将 Keras 模型转换为 C++

Question

已结束。此问题不符合 Stack Overflow guidelines .它目前不接受答案。

---

我们不允许提出有关书籍、工具、软件库等方面的建议的问题。您可以编辑问题，以便用事实和引用来回答它。

关闭3年前。

Improve this question

我正在使用 Keras(与 Theano)来训练我的 CNN 模型。有谁知道如何在我的 C++ 应用程序中使用它？有没有人尝试过类似的东西？我有想法编写一些 python 代码来生成具有网络功能的 c++ 代码 - 有什么建议吗？

我发现了一个类似的问题 here如何在 C++ 中使用 Tensorflow Keras 模型但没有答案。

Answer 1

为了回答我自己的问题并找到解决方案 - 我编写了一个名为 keras2cpp 的普通 C++ 解决方案(其代码可在 github 上找到)。

在此解决方案中，您存储网络架构(以 json 格式)和权重(以 hdf5 格式)。然后，您可以使用提供的脚本将网络转储到纯文本文件。您可以在纯 C++ 代码中将获得的文本文件与网络一起使用。不依赖于 python 库或 hdf5。它应该适用于 theano 和 tensorflow 后端。