spring98 / flutter-yolo-realtime-plugin

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

Error when running modified_export.py script. #3

Closed DenisMrGnusa closed 9 months ago

DenisMrGnusa commented 9 months ago

Hello, and thank you for your great plugin.

I have some issue when implement my own .pt file from roboflow. An error occurred when trying to run the modified_export.py script according to the instructions you wrote in the readme. However, I encountered an error as shown below:

Screenshot 2023-12-30 at 20 34 37
spring98 commented 9 months ago

hello! Thank you for your interest in my plugin.

It seems that the best_yolov5s.pt file is not in the same path as modified_export.py.

If the data was customized and trained, the pt file usually exists here: runs/train/exp/weights/best.pt

--weights runs/train/exp/weights/your_model.pt Please set the path like this.

DenisMrGnusa commented 9 months ago

thank you for your fast reply, I have managed to fix it, by changing the coremltools version to the same version as the one in your documentation (6.2).

jmtrutna commented 7 months ago

thank you for your fast reply, I have managed to fix it, by changing the coremltools version to the same version as the one in your documentation (6.2).

This worked for me as well.