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

Question

我训练了一个ANN模型，并将其保存为.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模型吗？

Answer 1

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

我猜您从模型优化器收到一个错误，因为您的模型未冻结。

有许多脚本可用于将Keras模型转换为冻结的Tensorflow图。例如，我可以推荐this one。

希望这会有所帮助。