时间戳

首页 > 解决方案 > 是否可以在 Java 中加载 TensorFlow 2.0 模型？

java - 是否可以在 Java 中加载 TensorFlow 2.0 模型？

问题描述

无法在谷歌上找到答案，这里提出的问题似乎都很接近但不一样。抱歉，如果我忽略了什么。

目前是否已经可以在 Java 8 中加载用 Python TensorFlow 2.0.0-beta1 编写的模型？该模型将使用 Keras Sequential API。如果这是可能的，我感谢指向相应文档的指针。

标签： javatensorflowtensorflow2.0

解决方案

如果您首先将模型转换为 tflite 文件，则可以完成此操作，如下所述：

import tensorflow as tf

converter = tf.lite.TFLiteConverter.from_keras_model_file("keras_model.h5")
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

然后您可以按照此处所述加载和运行：

public Interpreter(@NotNull File modelFile);

try (Interpreter interpreter = new Interpreter(file_of_a_tensorflowlite_model)) {
  interpreter.run(input, output);
}

推荐阅读