我正在尝试使用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 。有人可以帮我吗?项目的目录树是:
谢谢!