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

Question

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

My input and output tensors in python

Getting only shapes in android studio

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