onnx-modifier icon indicating copy to clipboard operation
onnx-modifier copied to clipboard

删除子分支后,分支上的节点变成了inputs。

Open ganleiboy opened this issue 1 year ago • 2 comments

1,原始模型删除子分支,如下图: screenshot_24

2,保存后打开修剪过的模型,可以看到inputs变多了,如下图: screenshot_25

3,我是通过如下代码对模型的inputs进行了删除:

import onnx                         

inputpath = "./descriptor.onnx"

# 1,修改模型
model = onnx.load(inputpath)
inputs = model.graph.input
inputs_num = 1  # 原始网络的输入会排在列表的最前面,所以调用pop函数删除冗余输入
for i in range(len(inputs)-inputs_num):
    inputs.pop()
print(inputs)

# 2,检查结构并保存模型(覆盖保存)
onnx.checker.check_model(model)
onnx.save(model, inputpath)

ganleiboy avatar Aug 08 '22 14:08 ganleiboy

你好,这个问题最近我在用的时候也发现了,正尝试修复,谢谢详细的反馈和提供备选解决方案!

ZhangGe6 avatar Aug 10 '22 14:08 ZhangGe6

@ganleiboy 你好,我更新了一个版本来尝试解决这个问题,如果方便的话,请使用新版做修改试试,如果仍然存在问题,欢迎继续讨论~

ZhangGe6 avatar Aug 12 '22 16:08 ZhangGe6