converted_model.tflite 未对有效的 TensorFlowLite 模型进行编码：无法打开 file_path。该模型不是有效的 Flatbuffer 文件

Question

我正在尝试从资产目录加载 converted_tflite.tflite。
它给我一个错误 java.lang.IllegalArgumentException: Contents of /file:/android_asset/converted_model.tflite does not encode a valid TensorFlowLite model: Could not open '/file:/android_asset/converted_model.tflite'.The model is not a valid Flatbuffer file

File file = new File("file:///android_asset/converted_model.tflite");
        try (Interpreter interpreter = new Interpreter(file)) {
            interpreter.run(inputData, output);
            Log.d("TF LOG", output);
        }catch(Exception e){
            e.printStackTrace();
}

我根据 Whosebug 的答案尝试了什么

aaptOptions {
    noCompress "tflite"
}

tensoflow-lite nighty 版

implementation 'org.tensorflow:tensorflow-lite:0.1.2-nightly'

Answer 1

如错误指示 model is not a valid Flatbuffer file。在您的实现中，模型是 File。它应该被转换为 flatbuffer 文件，如下所示，

FileInputStream f_input_stream= new FileInputStream(new File("file:///android_asset/converted_model.tflite"));
FileChannel f_channel = f_input_stream.getChannel();
MappedByteBuffer tflite_model = f_channel.map(FileChannel.MapMode.READ_ONLY, 0, f_channel .size());

然后您可以使用此 tflite_model 将 tflite 解释器创建为 New Interpreter(...)。

Answer 2

我使用了tensorflow-nighty构建gradle版本0.1.2

    implementation 'org.tensorflow:tensorflow-lite:0.1.2-nightly'

加载模型

/** Memory-map the model file in Assets. */
private static MappedByteBuffer loadModelFile(AssetManager assets, String modelFilename)
        throws IOException {
    AssetFileDescriptor fileDescriptor = assets.openFd(modelFilename);
    FileInputStream inputStream = new FileInputStream(fileDescriptor.getFileDescriptor());
    FileChannel fileChannel = inputStream.getChannel();
    long startOffset = fileDescriptor.getStartOffset();
    long declaredLength = fileDescriptor.getDeclaredLength();
    return fileChannel.map(FileChannel.MapMode.READ_ONLY, startOffset, declaredLength);
}

Answer 3

添加到现有答案中，如果您使用最新的 Tensorflow 版本 (2.4.0) 创建了 tflite 模型并面临类似问题，请将以下行添加到 build.gradle 文件[=12= 的依赖项中]

implementation 'org.tensorflow:tensorflow-lite:2.4.0'

并使用@LalitSharma 提供的函数从 'assets' 目录加载模型。

可以在这里找到最新版本 https://bintray.com/google/tensorflow/tensorflow-lite

converted_model.tflite 未对有效的 TensorFlowLite 模型进行编码：无法打开 file_path。该模型不是有效的 Flatbuffer 文件

converted_model.tflite does not encode a valid TensorFlowLite model: Could not open file_path. The model is not a valid Flatbuffer file

android

android-gradle-plugin

tensorflow

tensorflow-lite