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"}
}
)
onnx导出动态输入
最新推荐文章于 2024-05-20 20:36:51 发布