如何在tensorflow for java中使用tensorflow-lite模型

Question

是否可以使用tensorflow for java加载tensorflow lite模型？

我已经测试了SavedModleBundle和org.tensorflow.Graph.importGraphDef，但它不工作。

加载GraphDef时会出现java.lang.IllegalArgumentException: Invalid GraphDef异常。

看起来tflite解释器不是为tensorflow for java实现的。

诚挚的问候

Answer 1

要在独立的java (不是android)上使用tensorflow模型，您必须使用SavedModleBundle，并且需要按照here的说明使用java编译器进行编译。为此，您需要tensorflow提供的Java ( Jar )和TensorFlow Native Interface (JNI)文件。

不能在独立applications.Tensorflow中使用tflite模型。java精简版专门用于移动和嵌入式设备。