如何确保Tensorflow模型中的静态形状便于OpenVINO转换?

2024-09-29 06:34:52 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试优化并将tensorflow模型转换为OpenVINO IR。由于我在输入形状方面遇到的问题,它不是很成功。所以我打算用静态形状来重塑整个模型。我要研究的模型是keithito的Tacotron。你知道吗

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

将输入占位符节点设置为固定形状是否允许tensorflow推断和固定所有其他节点的形状?你知道吗


Tags: 模型节点irtensorflow静态形状重塑openvino
1条回答
网友
1楼 · 发布于 2024-09-29 06:34:52

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

Pythonmo_tf.py公司输入\u形状[1256256,3]输入\u模型型号.pb你知道吗

相关问题 更多 >