最近在做onnx转trt使用onnxsim出现这个错误,源代码如下:
from onnxsim import simplify
import onnx
input_path="cnn.onnx"
output_path="sim_cnn.onnx"
onnx_model = onnx.load(input_path) # load onnx model)
model_simp, check = simplify(onnx_model)
assert check, "Simplified ONNX model could not be validated"
onnx.save(model_simp, output_path)
print('finished exporting onnx')
加入--input-shape 后无异常,conv2d_input为输入name,可以用netron查看。
from onnxsim import simplify
import onnx
input_path="cnn.onnx"
output_path="sim_cnn.onnx"
onnx_model = onnx.load(input_path) # load onnx model
model_simp, check = simplify(onnx_model,input_shapes={'conv2d_input': [1, 240, 80, 3]})
assert check, "Simplified ONNX model could not be validated"
onnx.save(model_simp, output_path)
print('finished exporting onnx')