[Yolo] Exception: For an ML Program, extension must be .mlpackage (not .mlmodel) 에러 해결법

CoreML: export failure ❌ 8.9s: For an ML Program, extension must be .mlpackage (not .mlmodel). Please see https://coremltools.readme.io/docs/unified-conversion-api#target-conversion-formats to see the difference between neuralnetwork and mlprogram model types.

혹은

Traceback (most recent call last):
  File "export.py", line 863, in <module>
    main(opt)
  File "export.py", line 858, in main
    run(**vars(opt))
  File "/mnt/hdd_2T/IQA/ivan/yolov5/.venv/lib/python3.8/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
    return func(*args, **kwargs)
  File "export.py", line 778, in run
    pipeline_coreml(ct_model, im, file, model.names, y)
  File "export.py", line 694, in pipeline_coreml
    model.save(f)  # pipelined
  File "/mnt/hdd_2T/IQA/ivan/yolov5/.venv/lib/python3.8/site-packages/coremltools/models/model.py", line 459, in save
    raise Exception("For an ML Program, extension must be {} (not {})".format(_MLPACKAGE_EXTENSION, ext)

 

위와 같은 에러가 발생했다면 두가지 해결방법이 있습니다!

---

--nms 사용 안하는 경우

pip install coremltools==6.3.0 ultralytics

위와 같이 해결이 가능합니다.

 

--nms 사용 하는 경우

nms 사용시에는 6.2.0으로 다운그레이드 해야합니다.

pip uninstall coremltools

pip install coremltools==6.2.0 ultralytics

 

해결 되길 바랍니다!