Можно ли использовать оптимизатор моделей openVINO для преобразования моделей tenorflow и ann? - PullRequest
3 голосов
/ 23 января 2020

Я обучил модель ANN, сохранив ее как файл .h5. Затем я преобразовал модель в модель tenorflow и получил папку '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 моделей в первое место?

1 Ответ

2 голосов
/ 25 января 2020

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

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

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

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

...