728x90
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
해결 되길 바랍니다!
728x90
반응형
'Computer Vision' 카테고리의 다른 글
생체 인식 국제 표준 용어 정리 (1) | 2023.10.18 |
---|---|
Vision Transformer(ViT) 공부 및 간단 구현 예제 (0) | 2023.08.20 |
ResNet 구조를 이용한 AutoEncoder 구현 & 리뷰 (0) | 2023.08.16 |