【发布时间】：2018-07-05 08:17:40
【问题描述】：
我正在尝试将此模型：http://www.robots.ox.ac.uk/~vgg/data/vgg_face2/ 从 caffe 转换为 coreml，但出现此错误： RuntimeError：在 caffe 转换期间，“Reshape”类型的“conv2_1_prob_reshape”层中的参数“shape size”不支持选项“2”。
我使用了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
}
}
} 但我对此不太了解。
我在使用这些模型方面不是很有经验，感谢任何帮助。
【问题讨论】：
标签： python swift caffe coreml coremltools