import onnx input_path = "pp-docLayout-s-sim.onnx" output_path = "pp-docLayout-s-sim-cut.onnx" input_names = ["image"] output_names = ["Mul.109", "Concat.9"] onnx.utils.extract_model(input_path, output_path, input_names, output_names)