onnx删除无用属性

最新推荐文章于 2024-06-20 14:52:58 发布

牛andmore牛

最新推荐文章于 2024-06-20 14:52:58 发布

阅读量825

点赞数

分类专栏： ONNX 文章标签： python 人工智能开发语言

本文链接：https://blog.csdn.net/u011119817/article/details/128111258

版权

ONNX 专栏收录该内容

8 篇文章 1 订阅

订阅专栏

这里写自定义目录标题

在推理onnx模型时，报了一个错，如下：

InvalidGraph: [ONNXRuntimeError] : 10 : INVALID_GRAPH : This is an invalid model. In Node, ("Conv_0", Conv, "", -1) : ("x": tensor(float),"conv2d_68.tmp_0Weights": tensor(float),"conv2d_68.tmp_0Bias": tensor(float),) -> ("batch_norm_0.tmp_2": tensor(float),) , Error Unrecognized attribute: output_tensor_shape for operator Conv

unrecognized attribute,用netron 查看模型：
在这里插入图片描述
那么我们可以用以下代码对这类属性进行移除：

model = onnx.load('old.onnx')
nodes = model.graph.node
for node in nodes:
    attrs=node.attribute
    for attr in attrs[:]:
        if attr.name in ['excutor','output_tensor_type','output_tensor_shape']:
            attrs.remove(attr)
onnx.save(model,'new.onnx')