首页 文章

将caffe模型(VGG_Face2)转换为coreml时出错

提问于
浏览
1

我正在尝试将此模型转换为:http://www.robots.ox.ac.uk/~vgg/data/vgg_face2/从caffe到coreml但是我收到此错误:运行错误:不支持的选项'2',用于caffe转换期间'Reshape'类型的'conv2_1_prob_reshape'中的参数'shape size' .

我用 coreml_model = coremltools.converters.caffe.convert(('senet50_ft.caffemodel', 'senet50_ft.prototxt') . 我还尝试添加类标签(他们有一个包含四列的csv文件:ClassID,Name,Sample_Num,Flag . 我删除了最后两列,因为在其他教程中他们只使用前两列 . 我尝试了所有列,仅前两个以及原始的csv文件,而不将其转换为txt) . 以上所有都返回了相同的错误 . 我不确定这个错误意味着什么(我查看了包含所有图层的senet50_ft.prototxt以尝试找到错误,它看起来像这样: layer { name: "conv2_1_prob_reshape" type: "Reshape" bottom: "conv2_1_1x1_up" top: "conv2_1_prob_reshape" reshape_param { shape { dim: 0 dim: 0 } } } 但我没有't really understand much of this. I'在使用这些模型时没有经验,任何帮助非常感谢 .

1 回答

  • 1

    MLModel spec中的 ReshapeLayerParams 表示形状必须具有3或4维 . 您的原型文档仅提供2个维度(在 reshape_param 内) . 这就是给出这个错误的原因 .

相关问题