如何在Tensorflow模型中确保静态形状以轻松进行OpenVINO转换?

时间:2019-04-25 14:54:43

标签: python tensorflow speech openvino

我正在尝试优化并将张量流模型转换为OpenVINO IR。由于我在输入形状方面遇到的问题,它并不是很成功。因此,我计划使用静态形状重塑整个模型。我正在尝试使用的模型是keithito的Tacotron。

如何确保模型中的所有节点都具有静态形状?

仅将输入占位符节点设置为固定形状即可使张量流推断并固定所有其他节点的形状吗?

1 个答案:

答案 0 :(得分:0)

您可以在TF模型中具有动态形状,并在通过ModelOptimizer转换模型时提供静态形状。具有3个通道的256x256大小的输入数据示例。

python mo_tf.py --input_shape [1,256,256,3] --input_model model.pb