onnx.onnx_cpp2py_export.checker.ValidationError: Nodes in a graph must be topologically sorted的可能原因

onnx.onnx_cpp2py_export.checker.ValidationError: Nodes in a graph must be topologically sorted, however input 'model1_onnx::Conv_286' of node:
name: Conv_0 OpType: Conv
 is not output of any previous nodes.

上述报错的可能原因是,只修改了onnx中每个node的input,output属性和output的属性。而没有修改weight和bias ,running mean等参数,正确修改方法:


def rename_model(model, prefix='aaa'):
    for node in model.graph.node:
        node.name=f"{prefix}_{node.name}"

        for index in range(len(node.input)):
            node.input[index] = f"{prefix}_{node.input[index]}"
        for index in range(len(node.output)):
            node.output[index] = f"{prefix}_{node.output[index]}"


    for index in range(len(model.graph.initializer)):
        model.graph.initializer[index].name=f"{prefix}_{model.graph.initializer[index].name}"

    # rename input
    model.graph.input[0].name = f"vuy_img"
    model.graph.node[0].input[0] = f"vuy_img"

    # rename output
    for index in range(len(model.graph.output)):
        model.graph.output[index].name = f"{prefix}_{model.graph.output[index].name}"

可以通过调用下面的函数,看onnx格式是否正确:(如果不能被化简,则说明onnx格式不正确)

def simplify_onnx(onnx_path):
    """

    Args:
        onnx_path ():

    Returns:

    """
    onnx_model = onnx.load(onnx_path)  # load onnx model
    model_simp, check = simplify(onnx_model)
    assert check, "Simplified ONNX model could not be validated"

    save_path = onnx_path
    onnx.save(model_simp, save_path)
    print(f"Simplified onnx save path {save_path}")
    return save_path

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

York1996

您的打赏,是我更新的动力!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值