在 flutter 中加载自定义 ML 模型时出错

Question

我无法在 flutter 中加载 tensorflow lite 模型。我正在使用这个库

https://github.com/shaqian/flutter_tflite

当我尝试加载模型时，我收到了这个错误。

E/flutter ( 7258): [ERROR:flutter/lib/ui/ui_dart_state.cc(148)] Unhandled Exception: PlatformException(error, Unsupported value: java.io.FileNotFoundException: flutter_assets/assets/output.tflite,

我已按照库文档中的所有步骤进行操作。在没有这个库的情况下，是否有另一种方法可以在 flutter 中加载这个模型？

Answer 1

如果你已经训练了Keras模型(HDF5)，将其转换为Tensorflow模型(.pb)，然后将其转换为特莱特请参考以下代码：

转换器=tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)

tflite_model = converter.convert()

open("converted_model.tflite", "wb").write(tflite_model)

Answer 2

确保在您的 pubspec.yaml 文件中声明模型资产：

flutter:
  assets:
    - assets/tflite/model.tflite
    - assets/tflite/labels.txt

否则您将在尝试在代码中加载模型时收到此错误，例如：

      String tfResponse;
      tfResponse = await Tflite.loadModel(
        model: "assets/tflite/model.tflite",
        labels: "assets/tflite/labels.txt",

      );
      print("Response: TF Model Load $tfResponse");
    } catch (e) {
      print('Failed to load model. Error: $e');
    }

编辑：我正在使用这个包 https://pub.dev/packages/tflite

在 flutter 中加载自定义 ML 模型时出错

Error when load custom ML model in flutter

android

flutter

tensorflow

tensorflow-lite