Caffe to CoreML Model Conversion

0
votes

I have downloaded a model from the given link http://posefs1.perception.cs.cmu.edu/OpenPose/models/hand/pose_iter_102000.caffemodel

Then i use this Python code to convert this model into .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')

After compiling this code error is something like this

================= Starting Conversion from Caffe to CoreML ====================== Layer 0: Type: 'CPMData', Name: 'data'. Output(s): 'data', 'label'. Traceback (most recent call last): File "ModelConversionFile.py", line 2, in coreml_model = coremltools.converters.caffe.convert('pose_iter_102000.caffemodel','pose_deploy.prototxt') File "/Users/tahirhameed/Desktop/NewPythonTest/MyEnv/lib/python2.7/site-packages/coremltools/converters/caffe/_caffe_converter.py", line 191, in convert predicted_feature_name) File "/Users/tahirhameed/Desktop/NewPythonTest/MyEnv/lib/python2.7/site-packages/coremltools/converters/caffe/_caffe_converter.py", line 255, in _export predicted_feature_name) RuntimeError: Cannot convert caffe layer of type 'CPMData'.

1
caffecoreml

1 Answers

0
votes

The model you are trying to convert is containing a layer type (CPMData) that is not supported by CoreML. You would need to provide an implementation for that layer yourself.