Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

how to convert yolov8 custom model to tflite format #51

Open
chaukamm opened this issue Nov 6, 2024 · 2 comments
Open

how to convert yolov8 custom model to tflite format #51

chaukamm opened this issue Nov 6, 2024 · 2 comments

Comments

@chaukamm
Copy link

chaukamm commented Nov 6, 2024

i have tried convert .pt file to .onnx file, then convert to .pb file, finally .tflite file. but it seems didnt work

@LucasCarvalhoPAVIC
Copy link

I used this code in google colab to convert:

%pip install ultralytics
from ultralytics import YOLO
model = YOLO(r"/content/best.pt")
model.export(format="tflite")

@chaukamm
Copy link
Author

Ultralytics YOLOv8.0.151 Python-3.9.13 torch-2.5.1+cu118

i tried this to train model and it generate best.pt and i export to tflite
when i try to load the model in flutter, it shows this
[ERROR:flutter/runtime/dart_vm_initializer.cc(41)] Unhandled Exception: PlatformException(100, Error on load Yolov5 model, java.io.FileNotFoundException: flutter_assets/front2_new4.tflite
E/flutter ( 7059): at android.content.res.AssetManager.nativeOpenAssetFd(Native Method)
E/flutter ( 7059): at android.content.res.AssetManager.openFd(AssetManager.java:986)
E/flutter ( 7059): at com.vladih.computer_vision.flutter_vision.models.Yolo.initialize_model(Yolo.java:86)
E/flutter ( 7059): at com.vladih.computer_vision.flutter_vision.FlutterVisionPlugin.load_yolo_model(FlutterVisionPlugin.java:225)
E/flutter ( 7059): at com.vladih.computer_vision.flutter_vision.FlutterVisionPlugin.onMethodCall(FlutterVisionPlugin.java:99)
E/flutter ( 7059): at io.flutter.plugin.common.MethodChannel$IncomingMethodCallHandler.onMessage(MethodChannel.java:267)
E/flutter ( 7059): at io.flutter.embedding.engine.dart.DartMessenger.invokeHandler(DartMessenger.java:292)
E/flutter ( 7059): at io.flutter.embedding.engine.dart.DartMessenger.lambda$dispatchMessageToQueue$0$io-flutter-embedding-engine-dart-DartMessenger(DartMessenger.java:319)
E/flutter ( 7059): at io.flutter.embedding.engine.dart.DartMessenger$$ExternalSyntheticLambda0.run(Unknown Source:12)
E/flutter ( 7059): at android.os.Handler.handleCallback(Handler.java:942)
E/flutter ( 7059): at android.os.Handler.dispatchMessage(Handler.java:99)
E/flutter ( 7059): at android.os.Looper.loopOnce(Looper.java:226)
E/flutter ( 7059): at android.os.Looper.loop(Looper.java:313)
E/flutter ( 7059): at android.app.ActivityThread.main(ActivityThread.java:8810)
E/flutter ( 7059): at java.lang.reflect.Method.invoke(Native Method)
E/flutter ( 7059): at com.android.internal.os.RuntimeInit$MethodAndArgsCaller.run(RuntimeInit.java:604)
E/flutter ( 7059): at com.android.internal.os.ZygoteInit.main(ZygoteInit.java:1067)
E/flutter ( 7059): , null)
E/flutter ( 7059): #0 StandardMethodCodec.decodeEnvelope (package:flutter/src/services/message_codecs.dart:648:7)
E/flutter ( 7059): #1 MethodChannel._invokeMethod (package:flutter/src/services/platform_channel.dart:334:18)
E/flutter ( 7059):
E/flutter ( 7059): #2 AndroidFlutterVision.loadYoloModel (package:flutter_vision/src/plugin/android.dart:92:7)
E/flutter ( 7059):
E/flutter ( 7059): #3 _YoloImageV8SegState.loadYoloModel (package:flutter_vision_example/main.dart:708:5)
E/flutter ( 7059):
E/flutter ( 7059): #4 _YoloImageV8SegState.initState. (package:flutter_vision_example/main.dart:659:26)
E/flutter ( 7059):
E/flutter ( 7059):
I/ViewRootImpl@84c701eMainActivity: MSG_WINDOW_FOCUS_CHANGED 0 0

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants