Simple node deletion tool for onnx.
Project description
snd4onnx
Simple node deletion tool for onnx. I only test very miscellaneous and limited patterns as a hobby. There are probably a large number of bugs. Pull requests are welcome.
1. Setup
### option
$ echo export PATH="~/.local/bin:$PATH" >> ~/.bashrc \
&& source ~/.bashrc
### run
$ pip install -U onnx \
&& python3 -m pip install -U onnx_graphsurgeon --index-url https://pypi.ngc.nvidia.com \
&& pip install -U snd4onnx
2. Usage
$ snd4onnx -h
usage: snd4onnx [-h] onnx_file_path remove_node_names
positional arguments:
onnx_file_path
remove_node_names
optional arguments:
-h, --help show this help message and exit
3. Execution
$ snd4onnx input.onnx node_name_a
4. Sample
4-1. sample.1
Before | After |
---|---|
4-2. sample.2
Before | After |
---|---|
4-3. sample.3
Before | After |
---|---|
4-4. sample.4
Before | After |
---|---|
4-5. sample.5
Before | After |
---|---|
5. Reference
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
snd4onnx-0.0.2.tar.gz
(7.3 kB
view hashes)