onnx utils

其他资源 2025-08-02

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