无法在Google上找到答案,并且这里提出的问题似乎都很接近,但并不相同。对不起,如果我忽略了一些东西。

当前是否已经可以在Java 8中加载用Python TensorFlow 2.0.0-beta1编写的模型?该模型将使用Keras顺序API。如果可能的话,我感谢您提供相应文档的指针。

最佳答案

如果您首先按照here所述将模型转换为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)

然后可以按照here的说明加载并运行它:

public Interpreter(@NotNull File modelFile);

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

09-11 01:31