search

tensorflow / flutter-tflite

Apache License 2.0
426 stars 101 forks source link

Custom tflite model doesn't detect anything #226

Open fransay opened 2 weeks ago

fransay commented 2 weeks ago

Hi peeps,

I have been trying to train a custom model for object detection using this package, i.e flutter_tflite, but have found no success. custom models when ingested into the example app does not detect anything. Below are the approaches I have taken so far:

  1. Training with tflite_model_maker model_spec = 'efficientDET_lite0'

  2. Tensorflow 2 Object Detection API model_spec = 'ssd_mobilenet_v1/v2' && 'ssd_resnet50'

  3. Ultrayltics model_spec = yolo v8, v7, v5 -> output to a tflite.

None of the above has led to success, I am not able to perform detection on my honor x6a using the example app found here:https://github.com/tensorflow/flutter-tflite/tree/main/example/live_object_detection_ssd_mobilenet

Any help on instructions to train a custom working model for live detection use cases will be meet with a ton of gratitude and appreciation. Looking forward for help. Thanks.

GustavoSantiago113 commented 1 week ago

On the yolov8 model I have used the flutter vision package: https://pub.dev/packages/flutter_vision. I hope it works for you too

fransay commented 1 week ago

@GustavoSantiago113 , thank you for your reply! Flutter vision is definitely a very great package by all standards, only limitation is the IOS support is not yet ready. Let's hope Yuri or any other contributor has sometime to enable support in that direction. It stand tall among the best plugins for object detection in flutter. Again thank you very much for the suggestion.

darwinOne commented 3 days ago

Hi @fransay Did you find the solution? I had the same problem: not detecting anything with the yolov8 custom model.