python - 如何确保 Tensorflow 模型中的静态形状以便于 OpenVINO 转换？

Question

我正在尝试优化张量流模型并将其转换为 OpenVINO IR。由于我在输入形状方面面临的问题，它并不是很成功。所以我打算用静态形状改造整个模型。我正在尝试使用的模型是 keithito 的 Tacotron。

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

将输入占位符节点设置为固定形状是否允许 tensorflow 推断和修复所有其他节点的形状？

Answer 1

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

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