tensorflow_lite_flutter

一个 Flutter 项目，用于演示使用 teachablemachine 创建的 TensorFlow Lite 模型。

image2

image1

“TensorFlow”模型使用 Teachable Machines 进行训练。该模型通过不同的墙壁纹理颜色进行训练。应用将识别颜色并根据最佳匹配对颜色进行分类。此应用将加载预先训练好的模型，并在从相机控制器接收到的帧上开始分类。应用将实时显示结果以及置信度百分比。

可以从这个 下载链接 下载模型，并在“Teachable Machines”网站上重新加载。

标签

类别ID 标签名称
0 黑色纹理
1 蓝色纹理
2 绿色纹理
3 橙色纹理
4 粉色纹理
5 紫色纹理
6 红色纹理
7 白色纹理
8 黄色纹理

TFLite 辅助类

    // Load Model
    static Future<String> loadModel() async{
        AppHelper.log("loadModel", "Loading model..");
    
        return Tflite.loadModel(
          model: "assets/model_unquant.tflite",
          labels: "assets/labels.txt",
        );
    }
    
    //Start Classification on CameraImage frames
    static classifyImage(CameraImage image) async {
    
        await Tflite.runModelOnFrame(
                bytesList: image.planes.map((plane) {
                  return plane.bytes;
                }).toList(),
                numResults: 5)
            .then((value) {
      
          //Send results
          tfLiteResultsController.add(_outputs);
        });
    }

GitHub

https://github.com/umair13adil/tensorflow_lite_flutter