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

About the ultralytics hub with tflite file with custom dataset #7

Open
khchan246810 opened this issue Sep 17, 2024 · 6 comments
Open
Assignees
Labels
bug Something isn't working

Comments

@khchan246810
Copy link

Hi, I am the developer for Android Studio object detection. I know I am new to this function, and I might going to use ultralytics hub. Since, the txt file is not required if the model has metadata based on your code. I am not sure it can work by simply exporting the tflite model without needing txt file into the Android Studio. If it did detect the objects, the name might be not labelled or something.

@khchan246810 khchan246810 added the bug Something isn't working label Sep 17, 2024
@surendramaran
Copy link
Owner

If the metadata is not present in tflite then the ultralytics python library or its environment itself cannot label the classes.
So the default classes will be class0, class1, ...... class999

This repo will get the labels, for sure if metadata is presented.
Let me know if it doesn't.

@khchan246810
Copy link
Author

khchan246810 commented Sep 17, 2024

Ok thanks. Your YOLOv10 code got supported several YOLO versions such as YOLOv8, YOLOv9, and other versions? Or like YOLOv10 should have YOLOv10 tflite file, not other versions?

@surendramaran
Copy link
Owner

surendramaran commented Sep 17, 2024

Yeah, they should be yolov10 only for yolov10.

Yolov8 and yolov9 you can use yolov8 or yolov9 repo.

@khchan246810
Copy link
Author

Ok, thank you again.

@khchan246810
Copy link
Author

khchan246810 commented Sep 30, 2024

Hi sir, I want to ask for your opinion. What do you recommended for your custom dataset in terms of epoch, batch size, patience, and image size? Let's say if I have over 1000 images for 80 classes.

@moodpanda
Copy link

may I ask what did you use to export yolov8n.pt to yolov8.tflite? im using the ultralytics YOLO.export() but it does not seem to work. thank you so much for your response im really struggling working with yolov8

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

3 participants