search

ZhangGe6 / onnx-modifier

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

support feature "mark a tensor as model output" #13

Closed lordrebel closed 2 years ago

lordrebel commented 2 years ago

您好， 想问下可以支持选择一个node从的 输入或输出tensor 并把他标记成模型的输出吗？

ZhangGe6 commented 2 years ago

你好，请问这个需求是否和这里提到的相同？如果是的话，它已经被加入了todo list，不过暂时还不支持，您可以考虑手动调用ONNX Python API完成。

lordrebel commented 2 years ago

就是这个，谢谢～

ZhangGe6 commented 2 years ago

@lordrebel 你好，现在已经支持增加模型输出节点啦，欢迎使用~，详情参见readme。