loongson/pypi/: onnxsim0.4.36 metadata and description
Simplify your ONNX model
author  ONNX Simplifier Authors 
author_email  daquexian566@gmail.com 
classifiers 

description_content_type  text/markdown 
keywords  deeplearning ONNX 
license  Apache License v2.0 
requires_dist 

requires_python  >=3.7 
Because this project isn't in the mirror_whitelist
,
no releases from root/pypi are included.
File  Tox results  History 

onnxsim0.4.36cp37cp37mlinux_loongarch64.whl


ONNX Simplifier
ONNX is great, but sometimes too complicated.
Background
One day I wanted to export the following simple reshape operation to ONNX:
import torch class JustReshape(torch.nn.Module): def __init__(self): super(JustReshape, self).__init__() def forward(self, x): return x.view((x.shape[0], x.shape[1], x.shape[3], x.shape[2])) net = JustReshape() model_name = 'just_reshape.onnx' dummy_input = torch.randn(2, 3, 4, 5) torch.onnx.export(net, dummy_input, model_name, input_names=['input'], output_names=['output'])
The input shape in this model is static, so what I expected is
However, I got the following complicated model instead:
Our solution
ONNX Simplifier is presented to simplify the ONNX model. It infers the whole computation graph and then replaces the redundant operators with their constant outputs (a.k.a. constant folding).
Web version
We have published ONNX Simplifier on convertmodel.com. It works out of the box and doesn't need any installation. Note that it runs in the browser locally and your model is completely safe.
Python version
pip3 install U pip && pip3 install onnxsim
Then
onnxsim input_onnx_model output_onnx_model
For more advanced features, try the following command for help message
onnxsim h
Demonstration
An overall comparison between a complicated model and its simplified version:
Inscript workflow
If you would like to embed ONNX simplifier python package in another script, it is just that simple.
import onnx from onnxsim import simplify # load your predefined ONNX model model = onnx.load(filename) # convert model model_simp, check = simplify(model) assert check, "Simplified ONNX model could not be validated" # use model_simp as a standard ONNX model object
You can see more details of the API in onnxsim/onnx_simplifier.py
Projects Using ONNX Simplifier
 MXNet
 MMDetection
 YOLOv5
 ncnn
 ...
Chat
We created a Chinese QQ group for ONNX!
ONNX QQ Group (Chinese): 1021964010, verification code: nndab. Welcome to join!
For English users, I'm active on the ONNX Slack. You can find and chat with me (daquexian) there.