用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'))
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐