用coremltools6.0 转mlmodel的时候 出现了一些问题
报错1
Traceback (most recent call last):
File “”, line 1, in
AttributeError: module ‘coremltools.converters’ has no attribute ‘onnx’
原因:
coremltools6.0 没有了onnx转mlmodel
解决方法:
pip install coremltools==5
onnx 转 mlmodel code:
import os
import coremltools as ct
import onnx
import urllib
onnx_model_path = './xxx.onnx'
with open(onnx_model_path, 'rb') as f:
model_onnx = onnx.load(f)
cml_model = ct.converters.onnx.convert(
model=onnx_model_path,
minimum_ios_deployment_target='13',
)
cml_model.save('xxx.mlmodel')
报错2
onnx转mlmodel AttributeError: module ‘coremltools.converters’ has no attribute ‘caffe’
原因:
版本6.0, 5.0没有此功能
解决方法:
python3.7 pip install coremltools=4.0b2
pip install protobf==3.19
caffe转mlmodel code:
import os
import coremltools
import onnx
import urllib
coreml_model = coremltools.converters.caffe.convert(('xxx.caffemodel', 'xxx.prototxt'))