-
-
Notifications
You must be signed in to change notification settings - Fork 51
The model converted to coreml format always shows confidence 100 #87
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
Comments
👋 Hello @menghengmen, thank you for submitting a
In your case, it would be helpful to know:
Please make sure you've searched existing issues to avoid duplicates. If you need to add any additional information, please comment on this issue. 😊 An Ultralytics engineer will review and assist you soon. Thank you for helping us improve! 🚀 |
@menghengmen thanks for reporting this! To help us investigate, could you:
The CoreML export with NMS adds non-maximum suppression layers which might affect confidence score handling. For reference, see the CoreML export documentation section on NMS integration. |
![]() `)
3.when use nms = False ,load the mlmodel will crash at this // Retrieve class labels directly from the CoreML model's class labels, if available. |
Thanks for sharing the details! For the CoreML model.export(format="coreml", nms=False, names=['class1', 'class2', ...]) When using |
Python code
from ultralytics import YOLO
model = YOLO("yolo11n.pt")
model.export(format="coreml", nms=True, imgsz=640)
The text was updated successfully, but these errors were encountered: