仍然以【模型转化】修改onnx节点属性这篇文章中的例子为例来学习onnx的基本操作。这次不再修改Squeeze节点的属性(attribute),而是删除原有Squeeze节点,再在原来位置添加一个新的Squeeze节点。
1.定位要操作的节点
import onnx
model = onnx.load('my.onnx')
for node_id,node in enumerate(model.graph.node):
print("######%s######" % node_id)
print(node)
要操作的Squeeze为45号节点。
...
######45######
input: "384"
output: "385"
name: "Squeeze_250"
op_type: "Squeeze"
...
2. 删除并增加新节点
import onn

本文介绍如何在ONNX模型中删除原有的Squeeze节点,并在同一位置插入新的Squeeze节点,同时设置了新节点的axes参数。
最低0.47元/天 解锁文章

被折叠的 条评论
为什么被折叠?



