A set of simple tools for splitting, merging, OP deletion, size compression, rewriting attributes and constants, OP generation, and change opset for ONNX models.
Project description
simple-onnx-processing-tools
A set of simple tools for splitting, merging, OP deletion, size compression, rewriting attributes and constants, OP generation, and change opset for ONNX models.
1. Tools
$ pip install simple-onnx-processing-tools
$ docker run --rm -it \
-v `pwd`:/workdir \
-w /workdir \
pinto0309/simple-onnx-processing-tools:1.0.1
No. | Tool Name | Tags | Summary |
---|---|---|---|
1 | snc4onnx | Simple tool to combine(merge) onnx models. Simple Network Combine Tool for ONNX. | |
2 | sne4onnx | A very simple tool for situations where optimization with onnx-simplifier would exceed the Protocol Buffers upper file size limit of 2GB, or simply to separate onnx files to any size you want. Simple Network Extraction for ONNX. | |
3 | snd4onnx | Simple node deletion tool for onnx. Simple Node Deletion for ONNX. | |
4 | scs4onnx | A very simple tool that compresses the overall size of the ONNX model by aggregating duplicate constant values as much as possible. Simple Constant value Shrink for ONNX. | |
5 | sog4onnx | Simple ONNX operation generator. Simple Operation Generator for ONNX. | |
6 | sam4onnx | A very simple tool to rewrite parameters such as attributes and constants for OPs in ONNX models. Simple Attribute and Constant Modifier for ONNX. | |
7 | soc4onnx | A very simple tool that forces a change in the opset of an ONNX graph. Simple Opset Changer for ONNX. | |
8 | scc4onnx | Very simple NCHW and NHWC conversion tool for ONNX. Change to the specified input order for each and every input OP. Also, change the channel order of RGB and BGR. Simple Channel Converter for ONNX. |
2. Acknowledgments
- https://github.com/onnx/onnx/blob/main/docs/PythonAPIOverview.md
- https://docs.nvidia.com/deeplearning/tensorrt/onnx-graphsurgeon/docs/index.html
- https://github.com/NVIDIA/TensorRT/tree/main/tools/onnx-graphsurgeon
3. References
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for simple_onnx_processing_tools-1.0.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2ddcf9d0697d1cf51c2bae00edaf95883e22d9ef7363dcacee759dbba68f5243 |
|
MD5 | ed0d85698687200162cc6abf71683511 |
|
BLAKE2b-256 | 1689ce28d4f8e5862e487a632b6c10754e11eb1f0f5f6a90674028de1d188c63 |
Close
Hashes for simple_onnx_processing_tools-1.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e1b081b2625749a39e8562bdaa2af69e2769f92f701604acb8f31d86de9101fb |
|
MD5 | 5ec638706ed1c0e52cf5d102339d457d |
|
BLAKE2b-256 | 4c5efb49563cb5a5ad1325ee5eb66e34032ce767cf6ff2bc59f525f47cc04752 |