Как настроить coremltools преобразовать форму вывода модели, используя yolo.h5

Question

Я пытаюсь преобразовать h5 в iOS mlModel.

Я следую за Quick Start, чтобы получить 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, вам нужно будет заполнить форму вывода в spec.description.output используя coremltools.