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) | |