【模型转换】onnx删除并新增节点

本文介绍如何在ONNX模型中删除原有的Squeeze节点,并在同一位置插入新的Squeeze节点,同时设置了新节点的axes参数。

        仍然以【模型转化】修改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
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值