tensorflow 使用tf2onnx将模型从tf_hub转换为onnx

ktca8awb  于 2023-03-24  发布在  其他
关注(0)|答案(1)|浏览(197)

我从tensorflow hub下载了一个模型,然后尝试使用tf 2 onnx(https://github.com/onnx/tensorflow-onnx)将其转换为onnx。
然而,该模型在转换后似乎无法正常工作。
当我查看使用Netron生成的onnx文件时,输入层的维度为float 32 [unk__4256,unk__4257,unk__4258,3]。我相信这是由于输入模型的动态维度。然而,据我所知,这不一定是一个问题。
是否有任何方法可以覆盖模型的动态形状,并在转换过程中强制它具有特定的形状?
我环顾四周,看到其他人有类似的未知输入/输出层与其他模型,但它似乎工作无论如何。
提前感谢您的帮助!

czq61nw1

czq61nw11#

我发现了一个覆盖输入层形状的选项。命令应该是:

python -m tf2onnx.convert --saved-model model_path --output model.onnx --opset 17 --inputs 'input_0:0[16,100,100,3]'

其中input_0是输入层的名称。
希望这对其他人有帮助

相关问题