java - 从 Java 中的 tflite 模型进行推理

Question

我已导出 tflite 模型并在 this 上使用 Python 代码链接，我可以从这个模型中进行推断。然而，现在我正在尝试使用 Java 在 Android 应用程序中进行推理。我一直在关注官方文档here但我无法让它工作。有人可以指导我如何完成它吗？我所需要的只是执行以下步骤。

1. 读取 tflite 模型。
2. 生成形状为 [1,200,3] 的虚拟记录
3. 从 tflite 模型中进行推理并打印。

我一直在阅读 tflite 演示，但仍然无法解决它。要加载模型，我使用

Interpreter interpreter = new Interpreter(file_of_a_tensorflowlite_model)

从官方文档中得到以下错误:

error: no suitable constructor found for Interpreter(String)
constructor Interpreter.Interpreter(File) is not applicable
  (argument mismatch; String cannot be converted to File)

我无法解决这个问题。我该如何完成这个简单的任务？

Answer 1

您可以将 TFLite 模型粘贴到应用程序的 Assets 文件夹中。然后，使用此代码加载其 MappedByteBuffer。

 

private MappedByteBuffer loadModelFile() throws IOException {
        String MODEL_ASSETS_PATH = "recog_model.tflite";
        AssetFileDescriptor assetFileDescriptor = context.getAssets().openFd(MODEL_ASSETS_PATH) ;
        FileInputStream fileInputStream = new FileInputStream( assetFileDescriptor.getFileDescriptor() ) ;
        FileChannel fileChannel = fileInputStream.getChannel() ;
        long startoffset = assetFileDescriptor.getStartOffset() ;
        long declaredLength = assetFileDescriptor.getDeclaredLength() ;
        return fileChannel.map( FileChannel.MapMode.READ_ONLY , startoffset , declaredLength ) ;
}

然后在构造函数中调用它。

Interpreter interpreter = new Interpreter( loadModelFile() )