caffe - Caffe 到 CoreML 模型的转换

Question

我已经从给定的链接 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 行，在转换 predict_feature_name）文件“/Users/tahirhameed/Desktop/NewPythonTest/MyEnv/lib/python2.7 /site-packages/coremltools/converters/caffe/_caffe_converter.py"，第 255 行，在 _export predict_feature_name) RuntimeError：无法转换类型为“CPMData”的 caffe 层。

标签： caffecoreml

Answer 1

您尝试转换的模型包含CPMDataCoreML 不支持的层类型 ( )。您需要自己为该层提供一个实现。