Преобразование модели Caffe в CoreML - PullRequest
0 голосов
/ 28 февраля 2019

Я скачал модель по указанной ссылке 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 ', Имя:' данные '.Вывод (ы): «данные», «метка».Traceback (последний вызов был последним): файл «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 / convertters / caffe / _caffe_converter.py ", строка 191, в файле convert convertted_feature_name)" /Users/tahirhameed/Desktop/NewPythonTest/MyEnv/lib/python/site-packages/coremltools/converters/caffe/_caffe_converter.py ", строка 255, в _exportrealted_feature_name) RuntimeError: Невозможно преобразовать слой caffe типа 'CPMData'.

1 Ответ

0 голосов
/ 28 февраля 2019

Модель, которую вы пытаетесь преобразовать, содержит тип слоя (CPMData), который не поддерживается CoreML.Вам необходимо предоставить реализацию для этого уровня самостоятельно.

...