onnx-modifier使用

文章目录

**0、作用和功能**
**1、准备**
**2、界面介绍**
**3、注意tips**

onnx-modifier基于流行的模型可视化工具 Netron 和轻量级Web应用框架 flask 开发。希望它能给社区带来一些贡献

0、作用和功能

经常作为不同框架模型转化的中间文件，拿到ONNX文件，想将它进行一些修改
删除部分节点。比如，ONNX文件中一些前后处理的算子节点，以方便后续部署。
修改节点输入输出名。 比如修改某一节点的输入输出名称，更改模型拓扑结构。
编辑节点属性值。
增加新节点。

1、准备

直接拉取源代码：git clone git@github.com:ZhangGe6/onnx-modifier.git
进入目录 cd onnx-modifier
安装依赖包 pip install onnx flask
运行 python app.py
点击输出中的url（如http://127.0.0.1:5000/）

2、界面介绍

删除节点

Delete With Children 和 Delete Single Node.

修改节点的输入/输出名

通过修改节点的输出输出名，我们可以对模型拓扑结构进行修改（如删除一些预处理/后处理节点）。该功能同样可以用在更改模型的输出的名称（即修改模型叶子节点的输出名）。
-

编辑节点属性值

在节点侧边栏对应的属性值输入框中，键入新的属性值即可。

增加新节点

在selector中选择要添加的节点类型，在点击Add node按钮后，一个对应类型的新节点将自动出现在图上。
点击这个新节点，在弹出的侧边栏中进行节点的编辑
完工（点击Download即可获得编辑后的ONNX模型）。

3、注意tips

在当前版本中，是不支持添加含有参数的节点的（比如Conv, BatchNormalization）
为确保正确性，节点的各属性值建议全部填写（而不是留着undefined）
如果一个属性值是列表类型，则各元素之间使用‘,’分隔。

参考资料：https://blog.csdn.net/weixin_40561065/article/details/124554570?spm=1001.2014.3001.5502
参考资料：

onnx-modifier使用相关推荐

ONNX MLIR应用示例（含源码链接）
ONNX MLIR应用示例(含源码链接) 开放式神经网络交换在MLIR中的实现 (http://onnx.ai/onnx-mlir/). Prebuilt Containers 开始使用ONNX-ML ...
ONNX MLIR方法
ONNX MLIR方法 MLIR中的开放式神经网络交换实现. Prerequisites gcc >= 6.4 libprotoc >= 3.11.0 cmake >= 3.15.4 ...
使用Relay部署编译ONNX模型
使用Relay部署编译ONNX模型本文介绍如何使用Relay部署ONNX模型的入门. 首先,必须安装ONNX软件包. 一个快速的解决方案是安装protobuf编译器,然后 pip install o ...
使用ONNX将模型转移至Caffe2和移动端
使用ONNX将模型转移至Caffe2和移动端本文介绍如何使用 ONNX 将 PyTorch 中定义的模型转换为 ONNX 格式,然后将其加载到 Caffe2 中.一旦进入 Caffe2, 就可以运行 ...
ONNX 实时graph优化方法
ONNX 实时graph优化方法 ONNX实时提供了各种图形优化来提高模型性能.图优化本质上是图级别的转换,从小型图简化和节点消除,到更复杂的节点融合和布局优化. 图形优化根据其复杂性和功能分为几个类 ...
从PyTorch到ONNX的端到端AlexNet
从PyTorch到ONNX的端到端AlexNet 这是一个简单的脚本,可将Torchvision中定义的经过预训练的AlexNet导出到ONNX中.运行一轮推理Inference,然后将生成的跟踪模型 ...
用于ONNX的TensorRT后端
用于ONNX的TensorRT后端解析ONNX模型以使用TensorRT执行. 另请参阅TensorRT文档. 有关最近更改的列表,请参见changelog. 支持的TensorRT版本 Maste ...
编译ONNX模型Compile ONNX Models
编译ONNX模型Compile ONNX Models 本文是一篇介绍如何使用Relay部署ONNX模型的说明. 首先,必须安装ONNX包. 一个快速的解决方案是安装protobuf编译器,然后 pi ...
开放式神经网络交换-ONNX（下）
开放式神经网络交换-ONNX(下) 计算节点由名称.它调用的算子operator的名称.命名输入的列表.命名输出的列表和属性列表组成. 输入和输出在位置上与算子operator输入和输出相关联.属性通 ...
开放式神经网络交换-ONNX（上）
开放式神经网络交换-ONNX(上) 目的本文档包含ONNX语义的规范性规范. "onnx"文件夹下的.proto和.proto3文件构成了用协议缓冲区定义语言编写的语法规范..p ...