tflite

一個用于調(diào)用 TensorFlow Lite 的 Flutter 插件，支持 iOS 和 Android。

GitHub

Dart Pub

安裝

在 pubspec.yaml 文件中添加 tflite 依賴：

dependencies:
  tflite: ^0.0.2

Android

在 android/app/build.gradle 文件的 android 中添加以下設(shè)定：

    aaptOptions {
        noCompress 'tflite'
    }

用法

1. 創(chuàng)建 assets 文件夾，將 label 和 model 文件移至文件夾內(nèi)。在 pubspec.yaml 文件中添加：

  assets:
   - assets/labels.txt
   - assets/mobilenet_v1_1.0_224.tflite

2. 在代碼中引用庫：

import 'package:tflite/tflite.dart';

3. 加載 model 和 labels：

String res = await Tflite.loadModel(
  model: "assets/mobilenet_v1_1.0_224.tflite",
  labels: "assets/labels.txt",
);

4. 檢測圖片：

var recognitions = await Tflite.runModelOnImage(
  path: filepath,   // required
  inputSize: 224,   // wanted input size, defaults to 224
  numChannels: 3,   // wanted input channels, defaults to 3
  imageMean: 127.5, // defaults to 117.0
  imageStd: 127.5,  // defaults to 1.0
  numResults: 6,    // defaults to 5
  threshold: 0.05,  // defaults to 0.1
  numThreads: 1,    // defaults to 1
);

5. 釋放資源：

await Tflite.close();

示例代碼

使用 image_picker 插件獲取圖片，通過 tflite 插件調(diào)用 mobilenet 模型檢測圖片得出分類。

https://github.com/shaqian/flutter_tflite/tree/master/example

準備工作

下載 mobilenet_v1_1.0_224.tflite 并移至 ./assets 文件夾。

安裝

flutter packages get

運行

flutter run