A set of simple tools for splitting, merging, OP deletion, size compression, rewriting attributes and constants, OP generation, change opset, change to the specified input order, addition of OP, RGB to BGR conversion, change batch size, batch rename of OP, and JSON convertion 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, change opset, change to the specified input order, addition of OP, RGB to BGR conversion, change batch size, batch rename of OP, and JSON convertion for ONNX models.
1. Tools
HostPC
$ pip install -U simple-onnx-processing-tools \
&& pip install -U onnx \
&& python3 -m pip install -U onnx_graphsurgeon --index-url https://pypi.ngc.nvidia.com
Docker
$ docker run --rm -it \
-v `pwd`:/workdir \
-w /workdir \
pinto0309/simple-onnx-processing-tools:1.0.13
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. | |
9 | sna4onnx | Simple node addition tool for onnx. Simple Node Addition for ONNX. | |
10 | sbi4onnx | A very simple script that only initializes the batch size of ONNX. Simple Batchsize Initialization for ONNX. | |
11 | sor4onnx | Simple OP Renamer for ONNX. | |
12 | onnx2json | Exports the ONNX file to a JSON file. | |
13 | json2onnx | Converts a JSON file to an ONNX file. | |
14 | components_of_onnx | [WIP] | ONNX parts yard. The various operations described in Operator Schemas are converted in advance into OP stand-alone ONNX files. |
2. Very useful tools
No. | Tool Name | Author | Tags | Summary |
---|---|---|---|---|
1 | OnnxGraphQt | fateshelled | [WIP] | ONNX model visualizer. Model structure can be edited on the visualization tool. |
2-1. OnnxGraphQt - [WIP] Startup Method Sample
$ xhost +local: && \
docker run -it --rm \
-v `pwd`:/home/user/workdir \
-v /tmp/.X11-unix/:/tmp/.X11-unix:rw \
--net=host \
-e XDG_RUNTIME_DIR=$XDG_RUNTIME_DIR \
-e DISPLAY=$DISPLAY \
--privileged \
ghcr.io/pinto0309/openvino2tensorflow:latest
$ git clone https://github.com/fateshelled/OnnxGraphQt \
&& cd OnnxGraphQt \
&& sudo python3 -m pip install -r requirements.txt -U \
&& cd ..
$ python3 OnnxGraphQt/onnxgraphqt/main.py
3. 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
- https://github.com/onnx/onnx/blob/main/docs/Operators.md
4. 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.13.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 932a11b4069ab87c108f7f5949b1c5177a6ec593fc66af547b55226709b617c6 |
|
MD5 | aeee01fb02da7f2c1feb9f88406ef223 |
|
BLAKE2b-256 | 45993922751593330616768c1b1a1848421890210123c327b54f30629b724286 |
Close
Hashes for simple_onnx_processing_tools-1.0.13-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3fdeaff7bd39fb388915af0e53ecf498e2b274dcfde67ac9a95a1402fedabd06 |
|
MD5 | 6ff884d7e0cf875d512f5c275b68000d |
|
BLAKE2b-256 | 200c8574cf5ee0ae40285d5776f9c1d7009a44d62b85581bed48cafa9d4db50a |