Caffe到CoreML模型的转换

时间:2019-02-28 08:00:25

标签: caffe coreml

我已从给定链接下载了模型 http://posefs1.perception.cs.cmu.edu/OpenPose/models/hand/pose_iter_102000.caffemodel

然后我使用此Python代码将此模型转换为.mlmodel

导入coremltools coreml_model = coremltools.converters.caffe.convert('pose_iter_102000.caffemodel','pose_deploy.prototxt') coremltools.utils.save_spec(coreml_model,'my_model.mlmodel')

编译此代码后,错误是这样的

=================开始从Caffe到CoreML的转换====================== 第0层:类型:“ CPMData”,名称:“数据”。输出:“数据”,“标签”。 追溯(最近一次通话):   文件“ ModelConversionFile.py”,第2行,在     coreml_model = coremltools.converters.caffe.convert('pose_iter_102000.caffemodel','pose_deploy.prototxt')   文件“ /Users/tahirhameed/Desktop/NewPythonTest/MyEnv/lib/python2.7/site-packages/coremltools/converters/caffe/_caffe_converter.py”,第191行,已转换     预测特征名称)   _export中的文件“ /Users/tahirhameed/Desktop/NewPythonTest/MyEnv/lib/python2.7/site-packages/coremltools/converters/caffe/_caffe_converter.py”     预测特征名称) RuntimeError:无法转换“ CPMData”类型的Caffe层。

1 个答案:

答案 0 :(得分:0)

您要转换的模型包含CoreML不支持的图层类型(CPMData)。您需要自己provide an implementation进行该层设置。