search

surendramaran / YOLO

YOLOv8, YOLOv9, YOLOv10 in Mobile Devices, run different machine learning model inside Android and iOS.
https://www.surendramaran.com/
35 stars 12 forks source link

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

Open khchan246810 opened 1 week ago

khchan246810 commented 1 week ago

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.

surendramaran commented 1 week ago

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 commented 1 week ago

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 commented 1 week ago

Yeah, they should be yolov10 only for yolov10.

Yolov8 and yolov9 you can use yolov8 or yolov9 repo.

khchan246810 commented 1 week ago

Ok, thank you again.