Caffe到CoreML模型的转换

Question

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

然后，我使用这段Python代码将这个模型转换成.mlmodel

导入核心工具coreml_model = coremltools.converters.caffe.convert('pose_iter_102000.caffemodel'，'pose_deploy.prototxt') coremltools.utils.save_spec(coreml_model，'my_model.mlmodel')

编译此代码后，会出现类似如下的错误

=================开始从咖啡馆到CoreML ======================层0的转换:类型：'CPMData'，名称：'data‘。输出：'data'，'label‘。回溯(最近一次调用)：文件“"/Users/tahirhameed/Desktop/NewPythonTest/MyEnv/lib/python2.7/site-packages/coremltools/converters/caffe/_caffe_converter.py"，”，第2行，在coreml_model = coremltools.converters.caffe.convert('pose_iter_102000.caffemodel'，‘pose_deploy.prototxt’中)文件Traceback第191行，在convert "/Users/tahirhameed/Desktop/NewPythonTest/MyEnv/lib/python2.7/site-packages/coremltools/converters/caffe/_caffe_converter.py"，中)文件predicted_feature_name第255行，在_export predicted_feature_name中) RuntimeError:无法转换'CPMData‘类型的咖啡层。

Answer 1

您尝试转换的模型包含CoreML不支持的层类型(CPMData)。您需要自己为该层执行provide an implementation。