onnxslim_model_optimization_with_constant_folding_and_simplification.py

1import onnx
2from onnxslim import slim
3
4# Load your ONNX model
5model_path = "model.onnx"
6model = onnx.load(model_path)
7
8# Optimize the model
9# Slimming includes constant folding, shape inference, and removing unused nodes
10slimmed_model = slim(model)
11
12# Save the optimized model
13onnx.save(slimmed_model, "model_slimmed.onnx")