1

我有一个 .caffemodel 文件，我想通过 Caffe2Kit 在我的 iOS 应用程序中使用它，但实例初始化函数参数是 2 个名为“initNet”和“predictNet”的 .pb 文件。我尝试使用 caffe_translator：

python -m caffe2.python.caffe_translator deploy_nodist.prototxt global_model.caffemodel

但我收到一条错误消息：

KeyError: 'No translator registered for layer: name: "Slice"\ntype: "Slice"\nbottom: "data_l_ab_mask"\ntop: "data_l"\ntop: "data_ab_mask"\nslice_param {\n  slice_point: 1\n  axis: 1\n}\n yet.'

我还尝试使用 coremltools 将此 .caffemodel 文件转换为 .mlmodel 文件：

coreml_model = coremltools.converters.caffe.convert('global_model.caffemodel')

但我得到了这个：

Layer 0: Type: 'TransformingFastHDF5Input', Name: 'img'. Output(s): 'img'.
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/anaconda2/lib/python2.7/site-packages/coremltools/converters/caffe/_caffe_converter.py", line 191, in convert
    predicted_feature_name)
  File "/anaconda2/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 'TransformingFastHDF5Input'.

如何将此 .caffemodel 集成到我的 iOS 应用程序中？或者也许我需要为 mlmodel 使用自定义层？但是idk python。

4

0 回答 0