在 Java 中加载 keras 模型

数据挖掘 喀拉斯 爪哇
2021-10-02 05:58:24

Keras在 Java中加载经过训练的模型有什么要求?

我检查了DeepLearning4J支持Keras模型、网络架构和权重是否可以轻松加载。唯一的缺点可能是我们需要使用ND4J后端或者没关系?

如果有一个使用kerasand创建的模型tensorflow,加载它的最佳方法是什么Java ecosystem

我尝试使用冻结图形脚本来保存tensorflow模型,但无法使用SavedModel.load. 将模型加载为张量图对象似乎是一种常见的方式,但 Java 中的可用示例是混乱的,尤其是当输入是图像时。

1个回答

给你一些有效的链接:

deploying-keras-deep-learning-models-with-java:这使用了 DeepLearning4J,ND4J,据说你已经尝试过了。

张量流-keras-java

这里也提出了同样的问题——converting-keras-in-python-to-java它用KerasModelImport.importKerasSequentialModelAndWeight

您应该给出 HDF5 文件操作 java 示例:hdf-java如果您想自己更改权重,这将很有帮助

如果您已经尝试过所有这些,请发表评论。