import onnx
from onnxsim import simplify
onnx_model = onnx.load(“unet.onnx”) # load onnx model
model_simp, check = simplify(onnx_model)
print(‘ok’)
assert check, “Simplified ONNX model could not be validated”
output_path= r"C:\Users\320133514\Desktop\nnunetonnx\new.onnx"
onnx.save(model_simp, output_path)
print(‘finished exporting onnx’)