java - Tensorflow lite 模型中的输入和输出

Question

我已经训练了一个输入这个张量的模型：输入形状：[1 50] 输入类型：<class 'numpy.float32'>

并在输出这个张量：输出形状：[1 50 17] 输出类型：<class 'numpy.float32'>

我怎样才能在java中创建这些？在 Android 设备上使用它？谢谢

标签： javaandroidtensorflowkotlintensorflow-lite

Answer 1

有一些 TensorFlow Lite 语言绑定，包括用于 Android 应用程序开发的 Java 和 Kotlin。

• 官方 TensorFlow Lite Java API 可用。有关详细信息，请查看此链接。
• 最新版本的 Android Studio 还将 TensorFlow Lite 模型文件作为可接受的文件资产处理。