yolo(ptファイル)をcoreMLに変換してもラベルが追加されない場合や検知しない場合について

原因

YOLOの学習の結果では分類・枠線表示などが出来ているが、coreMLの変換はできるが上手くモデルが働いてくれない人向けです。

まずはxcodeなどでcoreMLを除いてclass labelsが設定されているか確認しよう

対処法

nms = True を付けよう 以上

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

変換するコードの全文

from ultralytics import YOLO
model = YOLO("your_model.pt") 
model.export(format='coreml', nms=True)