onnx-modifier

ONNX-Modifier 项目介绍

onnx-modifier 是一个专为 ONNX 模型编辑而设计的工具，旨在简化和加速模型编辑过程。传统上，编辑 ONNX 模型需要通过可视化模型图并借助 ONNX Python API 完成。这一过程不仅费时，而且每次修改后还需要重新进行可视化检查。为了解决这些问题，onnx-modifier 提供了一种全新的编辑方式，用户可以在完全可视化的界面中直接进行编辑，并即时预览编辑效果，大大提高了效率。

项目背景

ONNX 模型编辑的传统方式虽然功能强大，但由于需要编写代码并多次迭代检查结果，通常会消耗大量时间。onnx-modifier 通过将编辑过程与可视化预览相结合，简化了这一流程。用户可以在界面上专注于模型图的编辑，所有的编辑信息都会自动通过 Python ONNX API 处理和汇总，从而节省时间。

该工具基于流行的网络查看器 Netron 和轻量级的 Web 应用框架 Flask 构建，提供了一系列强大的功能来支持不同的编辑需求。

支持的编辑操作

onnx-modifier 支持多种编辑操作，其中包括：

删除节点
添加新节点
重命名节点的输入和输出
重命名模型的输入和输出
添加新的模型输出
添加新的模型输入
编辑模型输入形状
编辑节点属性
编辑模型的初始化器

使用指南

要使用 onnx-modifier，有以下几种方法：

通过命令行启动

首先克隆项目仓库并安装必需的 Python 包：

git clone https://github.com/ZhangGe6/onnx-modifier.git
cd onnx-modifier
pip install -r requirements.txt

然后运行：

python app.py

点击 Flaks 生成的输出信息中的 URL（通常为 http://127.0.0.1:5000/），onnx-modifier 将在 Web 浏览器中启动。

使用可执行文件启动

用户可以下载特定平台的可执行文件直接运行。在 Windows 平台上，可以从指定的网盘下载运行文件，双击即可启动。

通过 Docker 容器启动

也可以通过 Docker 容器来启动应用：

git clone git@github.com:ZhangGe6/onnx-modifier.git
cd onnx-modifier
docker build --file Dockerfile . -t onnx-modifier

构建完成后，映射 Docker 端口和本地文件夹来运行 onnx-modifier：

mkdir -p modified_onnx
docker run -d -t \
  --name onnx-modifier \
  -u $(id -u ${USER}):$(id -g ${USER}) \
  -v $(pwd)/modified_onnx:/modified_onnx \
  -p 5000:5000 \
  onnx-modifier