gpt4 book ai didi

c - 在 python 中使用 keras+tensorflow 进行训练并在 C/C++ 中进行预测

转载 作者:太空狗 更新时间:2023-10-29 11:49:25 27 4
gpt4 key购买 nike

是否可以使用以 tensorflow 为后端的 keras 制作一个快速模型,并使用它在 C 或 C++ 中进行预测?

我需要在 C++ 程序中进行预测,但我觉得在 keras 中进行模型和训练要舒服得多。

最佳答案

如果您不需要在要部署的环境中使用 GPU,您也可以使用我的库,称为 frugally-deep。它在 GitHub 上可用,并根据 MIT 许可证发布:https://github.com/Dobiasd/frugally-deep

frugally-deep 允许直接在 C++ 中对已训练的 Keras 模型运行前向传递,而无需链接到 TensorFlow 或任何其他后端。

它不仅支持使用顺序模型进行预测,还支持使用 functional API 构建的更复杂的模型进行预测.

除了支持许多常见的层类型外,它还可以跟上(有时甚至超过)TensorFlow 在单个 CPU 上的性能。您可以在 repo 中找到一些常见模型的最新基准测试结果。 .

通过自动测试 frugally-deep 保证在 C++ 中使用它的模型输出与在 Python 中使用 Keras 运行时完全相同。

关于c - 在 python 中使用 keras+tensorflow 进行训练并在 C/C++ 中进行预测,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46610922/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com