tensorflow - openVINO模型优化器可以用来转换tensorflow和nn模型吗？

Question

我训练了一个 ANN 模型并将其保存为 .h5 文件。然后我将模型转换为 tensorflow 模型并获得 '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 模型吗？

Answer 1

不幸的是，OpenVINO 不支持 Keras 模型格式。

我猜你从模型优化器中得到了一个错误，因为你的模型没有卡住。

有很多脚本可用于将 Keras 模型转换为卡住的 Tensorflow 图。我可以推荐this one例如。

希望能有所帮助。