一、前言
如标题,有几次朋友遇到这种情况,所以我想看看能不能直接更改 onnx
模型的 input shape
来解决这种问题。这种情况目前全发生在 tensorflow -> onnx
过程中,pytorch 由于有 onnx 的导出官方 api, 所以没有此烦心事。
二、代码与讲解
方法有两种,具体如下:
1、简化前修改输入 shape
先看看未修改前输入的无符号shape:
很明显,本该是1的地方出现了几个乱七八糟的符号。但是怎么做呢?我主要参考了这个。
from onnx.tools import update_model_dims
import onnx
def renew_shape():
onnx_file = "xxxx.onnx" # onnx path</