ONNX 动态batch_size导出

ONNX 动态batch_size导出

通过修改onnx图结构中输入输出的tensor尺寸来实现动态batch

import onnx
from onnx import shape_inference

model = onnx.load(r"FaceDetector.onnx")


input_info = model.graph.input[0]
input_name = input_info.name
input_shape = in