如何设置coremltools使用yolo.h5转换模型输出形状

Question

我尝试将h5转换为iOS mlModel。

我通过快速入门通过此链接获取h5文件。

https://github.com/qqwweee/keras-yolo3

然后，我使用coremltools将h5转换为mlModel

这是我的代码

mlModel = coremltools.converters.keras.convert('./yoloDone.h5',
                                            input_names='image',
                                            image_input_names='image',
                                            input_name_shape_dict={'image': [None, 416, 416, 3]})

但是我发现输出形状有些错误。

这是正确的输出。

这是我的输出，类型不相同。

正确的模型，output1为255x13x13。

但是我的模型，output1是1x1x255x13x13。

我该怎么做编辑？

Answer 1

1x1x255x13x13与255x13x13是相同的，除了您有5个尺寸而不是3个尺寸。

如果要使mlmodel输出255x13x13，则必须使用coremltools在spec.description.output中填写输出形状。