#183 · 主分类： 推理与本地部署
keras_to_tensorflow
deep-learning
keras
machine-learning
python
tensorflow
通用代码，将训练好的keras模型转换为推理用的tensorflow模型
项目最后更新：11/23/20
GitHub Stars
1.6K
Forks数量
524
贡献者数量
6
许可证
MIT
收录理由
不少团队在 Keras 里训练好模型，却要拿纯 TensorFlow 去做推理，中间总隔着一段别扭的空白。这个工具恰好补上这一步：它能把训练好的模型冻结成单个 .pb 文件，变量全部转成常量，推理图中没有贡献的死节点也会被顺手剪掉，最终得到的文件不再依赖原来的训练图就能加载。命令行参数覆盖了那些你本来要自己写脚本处理的部署细节，比如给多输出网络重命名输出张量、导出 checkpoint 和 metagraph 以便继续训练，或者对权重做量化。README 对自己的年头也很坦诚：它面向 TensorFlow 1.x 时代，并没有针对 2.0 做适配。如果你还在维护一条老 Keras 流水线，想从 .h5 走到冻结图并且步骤可重复，用它就能省下手写冻结样板代码的功夫。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容，欢迎 联系我们推荐选题或投稿。