File size: 329 Bytes
a725ece |
1 2 3 4 5 6 7 8 9 10 11 |
import onnx
input_path = "yolov8s.onnx"
output_path = "yolov8s-cut.onnx"
input_names = ["images"]
output_names = ["/model.22/Concat_output_0",
"/model.22/Concat_1_output_0",
"/model.22/Concat_2_output_0",]
onnx.utils.extract_model(input_path, output_path, input_names, output_names)
|