Можно ли использовать оптимизатор моделей openVINO для преобразования моделей ann tensorflow?

Я обучил модель ИНС, сохранив ее как файл .h5. Затем я преобразовал модель в модель тензорного потока и получил папки 'savedmodel.pb' и 'variables'. Затем я использовал оптимизатор моделей openvino для создания файлов IR, используя:

python3 mo_tf.py --input_model saved_model.pb

Но я получаю следующую ошибку:

[ FRAMEWORK ERROR ]  Error parsing message
TensorFlow cannot read the model file: "/home/user/Downloads/OpenVino/dldt-2019/model-optimizer/saved_model.pb" is incorrect TensorFlow model file

Можно ли openVINO использовать для преобразования ANN моделей в первую очередь?


tensorflow openvino inference-engine
person Fasty    schedule 23.01.2020    source источник

Ответы (1)


arrow_upward
2
arrow_downward

К сожалению, OpenVINO не поддерживает формат модели Keras.

Я предполагаю, что вы получили ошибку от оптимизатора модели, потому что ваша модель не заморожена.

Существует множество скриптов, которые можно использовать для преобразования модели Keras в замороженный граф Tensorflow. Я могу порекомендовать, например, этот.

Надеюсь, это поможет.

person Artemy Skrebkov    schedule 25.01.2020