4.3 cut onnx结构

前言

对现有onnx 进行剪切，得到部分onnx结构

Code

import onnx

org_name="new_sim.onnx"
# save_name="./example.onnx"
# save_name="./example2.onnx"
# save_name="./example3.onnx"
save_name="./example4.onnx"


input_name = ["input"]
# output_name = ["81"]
# output_name = ["83"]
# output_name = ["107"]
output_name = ["126"]

onnx.utils.extract_model(org_name, save_name, input_name, output_name)
print("DONE!!!")

总结

• 采用对应代码，可以进行剪切onnx
• 有些情况不能简单cut，需要进一步处理，这里不做介绍