onnx-utils
用于编辑或修改ONNX模型的实用脚本。脚本编辑并修改一个ONNX模型,以根据输入/输出节点名称和形状提取子图。
用法:onnx_edit.py [-h] [ - 输入输入] [ - 输出输出] [ - skipverify]输入输出
位置论点:
输入输入ONNX模型
输出输出ONNX模型
可选参数:
-h, - 赫尔普显示此帮助消息和退出
- 输入输入逗号分隔的模型名称附加了形状,例如 - 输入nodename1 [1,2,3],nodename2 [1,2,3]
- 输出输出逗号分隔的模型输出名称附加了形状,例如 - 输出OUTNODENAME1 [1,2,3],OUTNODENAME2 [1,2,3]
- 精通模型的跳过验证。如果形状不知道,有用
onnx-summarize
用法:onnx_summarize.py [-h]输入
在输入ONNX文件中创建一个操作员的摘要,并将每个循环主体倒入一个onnx文件,并为每个这样的循环主体创建摘要。
位置参数:输入输入ONNX模型
下载源码
通过命令行克隆项目:
git clone https://github.com/saurabh-shandilya/onnx-utils.git