File size: 292 Bytes
f400122
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
import onnx

input_path = "yolo11s.onnx"
output_path = "yolo11s-cut.onnx"
input_names = ["images"]
output_names = ["/model.23/Concat_output_0", "/model.23/Concat_1_output_0", "/model.23/Concat_2_output_0"]

onnx.utils.extract_model(input_path, output_path, input_names, output_names)