从caffemodel到coreml模型:添加input_shape后无法推断输入和分段错误的形状

时间:2018-12-01 00:31:13

标签: python caffe coremltools

我正在尝试使用coremltools将caffemodel转换为coreml模型。我的代码在下面:

import coremltools

model_path = '../models/'

coreml_model = coremltools.converters.caffe.convert((model_path + 'snapshots/itracker_iter_92000.caffemodel', model_path + 'itracker_deploy.prototxt', {'image_left': model_path + 'mean_images/mean_left_224.binaryproto', 'image_right': model_path + 'mean_images/mean_right_224.binaryproto', 'image_face': model_path + 'mean_images/mean_face_224.binaryproto'}), image_input_names=['image_left', 'image_right', 'image_face'])
coreml_model.save('eyes.mlmodel')

问题是,我收到此错误响应:

RuntimeError:无法推断输入“ facegrid”的形状。

在我的 deploy.prototxt 中,我有以下几行:

input: "facegrid"
input_dim: 256
input_dim: 625
input_dim: 1
input_dim: 1

在搜索了其他StackOverflow问题之后,我遵循了以下问题:Core ML model conversion fails with “Unable to infer input name and dimensions”,并在文件中添加了以下几行:

input: "facegrid"
input_dim: 256
input_dim: 625
input_dim: 1
input_dim: 1
input_shape {
  dim: 1
  dim: 1
  dim: 25
  dim: 25
}

但是现在我得到的只是一个细分错误:11 。有人可以帮我吗?项目的目录树是:

Directory Tree of the project

谢谢!

0 个答案:

没有答案