ZhangGe6 / onnx-modifier

A tool to modify ONNX models in a visualization fashion, based on Netron and Flask.
MIT License
1.28k stars 157 forks source link

关于算子属性的更改 #68

Closed Xu-Wade closed 1 year ago

Xu-Wade commented 1 year ago

我想请教下您基于netron的改动是什么,细节上是如何实现了算子属性的增减更改这一功能的?

ZhangGe6 commented 1 year ago

@Xu-Wade 主要是在netron可视化的基础上,增加了编辑的功能。

属性的修改逻辑是,在侧边栏捕获到用户对属性值的修改信息,传到后端(onnx_modifier.py),基于修改后的属性值重新生成一个新节点并替换原节点。具体可参见onnx_modifier.py中modify_node_attr函数