[CoreML] Custom Yolo v5 7.0 model returns error when trying to annotate

[Denizzje]

[CoreML] Custom Yolo v5 7.0 model returns error when trying to annotate

[Denizzje]

Hi @ryouchinsa ,

Thank you for your quick work. I did not even try with renaming as you already put the update in when I got access to the Macbook again.

I could succesfully load in a custom YoloV5 bounding box model, and a YoloV8 segmentation model. Surprised how fast it worked on a 2015 i5 Macbook. You have done a true great work there. We will get a M2 Mac mini now for labeling work with Rectlabel.

[ryouchinsa]

Thanks for writing the issue.
Please rename the Core ML model to yolov5.mlmodel.

RectLabel recognizes the model type through the file name and the short description. Such as YOLOv3, yolov5m, yolov5m-seg, yolov5m-cls, yolov8n, yolov8n-seg, yolov8n-cls, DeepLab, PoseNet, and FCRN-Depth.
https://rectlabel.com/help#load_coreml

[Denizzje]

Thank you, that was not clear to me from the documentation. I will try out next week when I have access to the macbook again.

To be clear, I both have yolov5s and yolov5m "size" models. They are different kind of models with a different purpose. The "m" model is a 25 class model. If I want to use that model, I still have to name it yolov5.mlmodel ?

Good to see you support yolov8 already by the way!

[ryouchinsa]

Improved Core ML menus so that RectLabel can load and run YOLOv5/v8 models when their file names do not include yolov5/v8 keywords such as best.mlmodel and last.mlmodel.

The new update version 2023.02.12 was released.
Please let us know your opinion.

In the previous version, the keyword "yolov5" was necessary in the file name to run the model as YOLOv5 model.
The number of classes can be calculated based on the output layer shape.
The number of classes = The output shape[2] - 5
Read this comment. ultralytics/yolov5#9485 (comment)