OnnxSlim can help you slim your onnx model, with less operators, but same accuracy, better inference speed.
pip install onnxslim
pip install git+https://github.com/inisis/OnnxSlim@main
git clone https://github.com/inisis/OnnxSlim && cd OnnxSlim/
pip install .
onnxslim your_onnx_model slimmed_onnx_model

import onnx
import onnxslim
model = onnx.load("model.onnx")
slimmed_model = onnxslim.slim(model)
if slimmed_model:
onnx.save(slimmed_model, "slimmed_model.onnx")
For more usage, see onnxslim -h or refer to our examples
Discord: https://discord.gg/nRw2Fd3VUS QQ Group: 873569894
Python
100.0%
OnnxSlim can help you slim your onnx model, with less operators, but same accuracy, better inference speed.
pip install onnxslim
pip install git+https://github.com/inisis/OnnxSlim@main
git clone https://github.com/inisis/OnnxSlim && cd OnnxSlim/
pip install .
onnxslim your_onnx_model slimmed_onnx_model

import onnx
import onnxslim
model = onnx.load("model.onnx")
slimmed_model = onnxslim.slim(model)
if slimmed_model:
onnx.save(slimmed_model, "slimmed_model.onnx")
For more usage, see onnxslim -h or refer to our examples
Discord: https://discord.gg/nRw2Fd3VUS QQ Group: 873569894
Python
100.0%