| import onnx | |
| input_path = "yolov8s-seg.onnx" | |
| output_path = "yolov8s-seg-cut.onnx" | |
| input_names = ["images"] | |
| output_names = ["/model.22/Concat_1_output_0", | |
| "/model.22/Concat_2_output_0", | |
| "/model.22/Concat_3_output_0", | |
| "/model.22/cv4.0/cv4.0.2/Conv_output_0", | |
| "/model.22/cv4.1/cv4.1.2/Conv_output_0", | |
| "/model.22/cv4.2/cv4.2.2/Conv_output_0", | |
| "output1"] | |
| onnx.utils.extract_model(input_path, output_path, input_names, output_names) | |