search

tucan9389 / ObjectDetection-CoreML

An example running Object Detection using Core ML (YOLOv8, YOLOv5, YOLOv3, MobileNetV2+SSDLite)
https://github.com/motlabs/awesome-ml-demos-with-ios
MIT License
297 stars 51 forks source link

Yolo8 to coreml conversion #11

Open dhiman10 opened 1 year ago

dhiman10 commented 1 year ago

from ultralytics import YOLO

model=YOLO('runs/detect/train/weights/best.pt')

model.export(format='coreml',nms=True)

CoreML: export failure ❌ 9.9s: 1 name found for nc=8395 CoreML model is not converted.

I have found this error when I use nms=True, I need the model with bounding box. Could you help me ?