我正在使用Keras(与Theano)来训练CNN模型。有谁知道如何在我的C ++应用程序中使用它?有人尝试过类似的东西吗?我有想法写一些python代码,它将生成具有网络功能的c ++代码-对此有何建议?
我在这里找到了类似的问题,如何在C ++中使用Tensorflow Keras模型但没有答案。
为了回答我自己的问题并找到解决方案-我编写了一个简单的c ++解决方案keras2cpp(其代码可在github上找到)。
在此解决方案中,您将存储网络体系结构(以json格式)和权重(以hdf5格式)。然后,您可以使用提供的脚本将网络转储到纯文本文件。您可以使用纯c ++代码在网络上使用获得的文本文件。python库或hdf5没有依赖关系。它应该适用于theano和tensorflow后端。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
点击生成二维码
我来说两句