Caffe到CoreML模型转换

问题描述 投票:0回答:1

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

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

import 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 prediction_feature_name中)RuntimeError:无法转换'CPMData'类型的caffe图层。

caffe coreml
1个回答
0
投票

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

© www.soinside.com 2019 - 2024. All rights reserved.