spring98 / flutter-yolo-realtime-plugin

실시간 객체검출 플러그인 배포 (2023)
MIT License
7 stars 4 forks source link

GET THE MODEL #1

Open zanndev0511 opened 9 months ago

zanndev0511 commented 9 months ago

May I ask where did you get this model? image

spring98 commented 9 months ago

hello!

Thank you for using my plugin.

First of all, I would like to apologize for the late reply.

In essence, if you want to use the model on iOS, you must run modified_export.py to obtain the model as specified in readme.md. This is because you can obtain class label information together.

If you use export.py from the existing yolo repository, class label information will not appear.

How to use

python modified_export.py --weights your_weight.pt --include coreml

You can use it like this.

thank you