pip install -r requirements.txt
python make_onnx_dynamic_shape_to_fixed.py --onnx_path yolov5n.onnx --input_name 'images' --input_shape 6,3,640,640
python change_batch_size_for_static_shape.py --onnx_path yolov5n_bs6.onnx --batch_size 2
Change batch size for static shape/dynamic shape onnx model
pip install -r requirements.txt
python make_onnx_dynamic_shape_to_fixed.py --onnx_path yolov5n.onnx --input_name 'images' --input_shape 6,3,640,640
python change_batch_size_for_static_shape.py --onnx_path yolov5n_bs6.onnx --batch_size 2
Change batch size for static shape/dynamic shape onnx model