Skip to content

A flutter project for demonstarting usage of TensorFlow Lite model created with teachablemachine.

Notifications You must be signed in to change notification settings

joaorenatojr87/tensorflow_lite_flutter

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

tensorflow_lite_flutter

A flutter app to demonstrate usage of tensor flow lite ML.

App Demo Teachable Machine

The "TensorFlow" model is trained using Teachable Machines. The model is trained with different texture colors of walls. App will recognize the color and classify the color according to best match. This app will load a pre-trained model and start classification on frames received from Camera Controller. App will show results in real-time along with confidence percentages.

Model can be downloaded from this download link and re-loaded on "Teachable Machines" website.

Labels

Class Id Label Name
0 Black Texture
1 Blue Texture
2 Green Texture
3 Orange Texture
4 Pink Texture
5 Purple Texture
6 Red Texture
7 White Texture
8 Yellow Texture

TFLite Helper Class

    // 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);
        });
    }

Getting Started

This project is a starting point for a Flutter application.

A few resources to get you started if this is your first Flutter project:

For help getting started with Flutter, view our online documentation, which offers tutorials, samples, guidance on mobile development, and a full API reference.

About

A flutter project for demonstarting usage of TensorFlow Lite model created with teachablemachine.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Dart 70.1%
  • Ruby 23.8%
  • Kotlin 3.0%
  • Swift 2.8%
  • Objective-C 0.3%