使用openvino模型优化器将保存的张量流模型转换为IR

时间:2020-07-21 22:08:55

标签: tensorflow openvino

我使用keras和tensorflow 2.2.0作为背景训练了定制的CNN模型。之后,我将模型另存为.ckpt文件,其中包含资产,变量,.pb文件作为子文件夹init。之后,将其转换为openvino文档中的IR,可以使用以下命令:

**要转换这样的TensorFlow模型: 转到 / deployment_tools / model_optimizer目录

运行带有指向SavedModel目录的路径的mo_tf.py脚本以转换模型:

python3 mo_tf.py --saved_model_dir **

所以,我如上所述进入了以下目录并厌倦了以下命令:

python3 mo_tf.py --saved_model_dir C:\ Users \ vyas \ Desktop \ saved_model \ cp.ckpt

没有输出或任何东西。也没有错误。 另外,我尝试了以下命令:

python3 mo_tf.py --saved_model_dir C:\ Users \ vyas \ Desktop \ saved_model \ cp.ckpt --output_dir C:\ Users \ vyas \ Desktop \ out

仍然没有输出。

有人可以帮忙吗?

我正在使用tensorflow 2.2.0

有人可以帮我吗

1 个答案:

答案 0 :(得分:1)

-saved_model_dir必须提供SavedModel目录的路径。

按如下所示修改您的命令:

python3 mo_tf.py --saved_model_dir C:\ Users \ vyas \ Desktop \ saved_model