代码之家 › 专栏 › 技术社区 › gustavz

如何处理org.tensorflow.lite.interpreter.runformultipleinputsoutputs()的结果

tensorflow-lite tensorflow android-studio arrays java

gustavz · 技术社区 · 6 年前

我用tflite在android上运行posenet(cnn)。模型有多个输出数组,具有以下维度: 1x14x14x17, 1x14x14x34, 1x14x14x32, 1x14x14x32

因此,运行Java TFLITE解释器

import org.tensorflow.lite.Interpreter;
Interpreter tflite;
...
tflite.runForMultipleInputsOutputs(inputs,outputs)

我可以访问四个输出张量。 tflite.getOutputTensor(i) 或与 outputs.get(i) (和我一起。[0,3]) outputs 是一个 HashMap 充满 java.nio.HeapByteBuffer 物体。

如何将这些输出或子矩阵张量转换成Java多维数组(类似于 float[][][][]; )能够对它们进行数学计算?

0 回复 | 直到 6 年前

gustavz 6 年前

定义如下的输出使您可以使用原生Java数组,这是我想要的:

out1 = new float[1][14][14][17];
out2 = new float[1][14][14][34];
out3 = new float[1][14][14][32];
out4 = new float[1][14][14][32];
Map<Integer, Object> outputs = new HashMap<>();
outputs.put(0, out1);
outputs.put(1, out2);
outputs.put(2, out3);
outputs.put(3, out4);

推荐文章

somnathchakrabarti · 从Kotlin 2.0开始,启用Compose时需要Compose编译器Gradle插件

7 月前

QuantamPulse · Android Studio教程的问题=添加图像

1 年前

BobDoolittle · 使用Room开发Android应用程序-为什么在构建之间清除数据库?

1 年前

juske · 将方法作为可点击lambda中的参数传递给Composable。该表达式未使用

1 年前

Jeremi · 如何只获得一些羽毛工具栏按钮?

1 年前

Surbhi Gupta · 在将firebase testlab集成到我在webdriverIO和cumber框架中开发的本地移动自动化项目中时遇到的问题

1 年前

codproe · 安装Android后我无法打开应用程序

1 年前

olistocks98 · Android Studio中Jetpack Compose的代码自动完成无法正常工作

1 年前

Hi Mr · 如何在Android Studio中实现各种功能:Intents、AlertDialog、NotificationManager、MediaPlayer和SQLite数据库操作?

1 年前

Shervin.bdn · Flutter中的完全可滚动页面问题

1 年前