File size: 540 Bytes
17da060
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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)