【问题标题】:converting keras in python to java将python中的keras转换为java
【发布时间】:2019-01-09 03:17:42
【问题描述】:

我目前在 python 中使用 keras 有一个工作神经网络,但是,我需要最终产品在 java 中。到目前为止,我的研究还没有显示 keras 的 python 和 java 之间的直接转换,但是有没有办法在 python 中使用 keras 保存模型,然后就像我们在 python 的另一个文件中加载模型一样,在 java 中加载模型?或者换句话说,将我的 keras 神经网络转换为与 java 兼容的形式的最简单方法是什么?

【问题讨论】:

    标签: java python neural-network keras


    【解决方案1】:

    DeepLearning4J 可用于加载 keras 模型。

    如果您同时拥有 json 文件和 .h5(weights) 文件,您可以使用KerasModelImport.importKerasSequentialModelAndWeights 类来加载模型,如下所示。

    MultiLayerNetwork network = KerasModelImport.importKerasSequentialModelAndWeights("PATH TO YOUR JSON FILE","PATH TO YOUR H5 FILE")
    

    【讨论】:

      【解决方案2】:

      您可以使用 H5 文件格式传输权重和架构。 Keras推荐h5。

      https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model

      【讨论】:

        【解决方案3】:

        现在您可以使用Deep Java Library (DJL) 在 Java 中加载 Keras 模型并运行推理。 DJL 是一个开源的 Java 深度学习库。查看 github 仓库:https://github.com/awslabs/djl

        有一篇博文:https://towardsdatascience.com/detecting-pneumonia-from-chest-x-ray-images-e02bcf705dd6

        并且可以找到演示项目:https://github.com/aws-samples/djl-demo/blob/master/pneumonia-detection/README.md

        【讨论】:

          猜你喜欢
          • 2019-10-13
          • 1970-01-01
          • 2014-05-14
          • 2018-06-11
          • 2021-07-10
          • 2016-12-14
          • 1970-01-01
          • 2018-07-13
          • 1970-01-01
          相关资源
          最近更新 更多