更改ONNX模型的输入大小

Question

我需要将ONNX模型的输入大小从[1024,2048,3]更改为[1,1024,2048,3]

为此，我尝试使用ONNX的update_inputs_outputs_dims

import onnx
from onnx.tools import update_model_dims

model = onnx.load("./0818_pspnet_1.0_713_resnet_v1/pspnet_citysc.onnx")
updated_model = update_model_dims.update_inputs_outputs_dims(model, {"inputs:0":[1,1024,2048,3]}, {"predictions:0":[1, 1025, 2049, 1]})
onnx.save(updated_model, 'pspnet_citysc_upd.onnx')

然而，这是我最后的错误

ValueError: Unable to set dimension value to 1 for axis 0 of inputs:0. Contradicts existing dimension value 1024.

ONNX模型是从PSPNet的Tensorflow冻结图导出的。如果上述方法不起作用，我是否需要修改冻结图

非常感谢您的帮助