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

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 图。例如，我可以推荐这个。

希望它会有所帮助。