android - 如何在android studio中读取输出张量（tflite）

Question

我可以在 python 中获取形状并读取张量本身，但只能在 android tflite 中使用 tflite.getOutputTensor(0).shape() 和 tflite.getInputTensors(0).shape() 获取张量的形状。我在 python 中使用 .get_tensor(nodevalue) 读取张量，但我应该如何在 android 中执行此操作

我在 python 中的输入和输出张量

在 android studio 中只获取形状

标签： androidtensorflowtensorflow-lite

Answer 1

您必须在调用之前将输出缓冲区分配给解释器。有关示例代码，请参见以下链接。

https://github.com/tensorflow/examples/blob/ee198ec873148a5df3fd6eedb8d396613164ead7/lite/examples/image_classification/android/lib_support/src/main/java/org/tensorflow/lite/examples/classification/tflite/Classifier.java#L258

或者对于多个输出，

https://github.com/tensorflow/examples/blob/ee198ec873148a5df3fd6eedb8d396613164ead7/lite/examples/object_detection/android/lib_interpreter/src/main/java/org/tensorflow/lite/examples/detection/tflite/TFLiteObjectDetectionAPIModel.java#L202