Skip to content

Utility scripts for editing or modifying onnx models

License

Notifications You must be signed in to change notification settings

MohitIntel/onnx-utils

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 

Repository files navigation

onnx-utils

Utility scripts for editing or modifying onnx models. The script edits and modifies an onnx model to extract a subgraph based on input/output node names and shapes.

usage: onnx_edit.py [-h] [--inputs INPUTS] [--outputs OUTPUTS] [--skipverify] input output

positional arguments:

input input onnx model
output output onnx model

optional arguments:

-h, --help show this help message and exit

--inputs INPUTS comma separated model input names appended with shapes, e.g. --inputs nodename1[1,2,3],nodename2[1,2,3]

--outputs OUTPUTS comma separated model output names appended with shapes, e.g. --outputs outnodename1[1,2,3],outnodename2[1,2,3]

--skipverify skip verification of model. Useful if shapes are not known

About

Utility scripts for editing or modifying onnx models

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%