onnx导出动态输入_onnx动态输入shape

x = preprocess(img_path)



    # 导出ONNX模型
    torch.onnx.export(
        model=model,
        args=x,
        f="model.onnx",
        input_names=["input"],
        output_names=["output"],
        dynamic_axes={
            "input": {0: "batch_size", 2: "height", 3: "width"},
            "output": {0: "batch_size"}
        }
    )
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16