Flutter 图像分类

此应用可根据任何 TFLite 图像分类模型对图像进行分类。
已提供一个用于分类猫狗的示例模型。

Alt text

使用您的 TFLite 模型说明

  1. 安装 Flutter 并克隆仓库。
  2. 修改 assets/tflite/[name].tflite 中的模型。
  3. 修改 assets/tflite/[name].txt 中的类名以表示您的类（见下方示例）。
  4. 在 pubspec.yaml 中修改 [name].tflite 和 [name].txt（见下方示例）。
  5. 更改 lib/main.dart 中的 loadModel 函数以加载您的 [name].tflite 和 [name].txt（见下方示例）。
  6. 运行 Flutter 应用。tflite 包使用 v1 嵌入，因此我们需要传递 –ignore-deprecation。
flutter run --ignore-deprecation

cats_dogs.txt

Cat
Dog

pubspec.yaml

assets {
    - assets/tflite/[name].tflite
    - assets/tflite/[name].txt
}

lib/main.dart

String res = await Tflite.loadModel(
    model: "assets/tflite/[name].tflite",
    labels: "assets/tflite/[name].txt",
) ?? '';

GitHub

查看 Github