netron
netron copied to clipboard
lutzroeder
Model editing support
Is there any plan to have Netron support model editing please? This will be a very useful feature
There's one more scenario happened frequently is, do some manual constant folding optimization. Say, it's known that a sub-graph can be constant-folded, and user would like to directly remove the sub-graph and replace it with an initializer.
@lutzroeder Editing would be great. I really would like to use it for training with yolo and darknet. In fact I found your tool googleing exactly for a way to edit the files with a gui
Editing feature in Netron itself would be very cool!! I love the Netron UI for ONNX model inspection and all information is presented in one integrated fashion!
Even just basic editing support would be amazing:
- Include/exclude nodes from the list of input/output nodes
- Remove nodes
Since models can be edited through the onnx python API e.g here, it would be a possible way of resolving this to introduce a way to run python script, and then the community could add their own scripts into the editor.
This could be helpful for anyone looking for something similar: https://github.com/ZhangGe6/onnx-modifier It builds on top of Netron.
Interested in this feature too and like to contribute. It involves 3 areas:
- side-bar attribute view need to be editable
- data changed in side-bar be accessiable by model
- serialization (protobuf, json, flatbuffer, xml...)
The current protoc.js and protobuf.js only have decoder/reader. Should we add encoder/writer by ourself? Or should we use existing project like https://github.com/protobufjs/protobuf.js/ ?
Should we add encoder/writer by ourself?
Yes, extend the existing libraries instead of adding dependencies.
I have an initial implementation for protobuf encoding (https://github.com/lutzroeder/netron/pull/995)
This work is still active and could be help
https://github.com/ZhangGe6/onnx-modifier
