ONNX修改节点的值

#新节点
conv1_node = onnx.helper.make_node(
    name="Conv_40",  # Name is optional.
    op_type="Conv",
    # Must follow the order of input and output definitions.
    # https://github.com/onnx/onnx/blob/rel-1.9.0/docs/Operators.md#inputs-2---3
    inputs=["input", "backbone.patch_embed.proj.weight", "backbone.patch_embed.proj.bias"],
    outputs=["268"],
    # The following arguments are attributes.
    kernel_shape=[4,4,4],
    strides=[4,4,4],
    dilations=[4,4,4],
    group=1,
    pads=[0,0,0,0,0,0]
)


onnx_model = onnx.load(onnxfile)
graph = onnx_model.graph
graph.node.remove(node)
graph.node.insert(584, conv1_node)
onnx.checker.check_model(onnx_model)
onnx.save(onnx_model, onnxfile)

参考文章:
https://leimao.github.io/blog/ONNX-Python-API/
https://github.com/onnx/onnx/issues/2259

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值