onnx_tensorrt报错:Attribute not found: height_scale

在使用onnx_tensorrt转换onnx为tensorrt格式的模型时报错Attribute not found: height_scale。

如果onnx模型是通过pytorch 用插值上采样转换来的,参考:
https://www.jianshu.com/p/dcd450aa2e41

由于我的onnx模型是yolov3-tiny转换的模型,不适用。
由于github项目下给出了两种方法,一种通过backend来转换。另一种不通过backend,直接使用onnx2trt转换。
我在使用darkend转换时报错,又试了一下后面的方法,结果竟然成功了,在这里记录一下。

第一种方法:
在这里插入图片描述报错:

在这里插入图片描述
第二种方法:
在这里插入图片描述先进入到你放onnx模型的文件夹,执行:
在这里插入图片描述成功!
在这里插入图片描述转换失败时也会生成trt模型,但是是个空的文件,所以要留意。

除此之外,还要注意一下:
onnx转换成tensorrt模型前,请先确保转换出来的onnx模型是能用的。比如拿一张图片识别一下,看看是否能成功识别。
有些模型能成功转换为onnx,但是转换完以后是不能用的,如果想要把不能用的onnx模型转换为tensorrt模型,就有可能会出现以下错误:
在这里插入图片描述报错:Assertion failed: tensors.count(output.name())
这是的yolov3模型出错,转换成功了,但是测试中是不能用的,转trt也失败了。

更新于2020.11.16
报这个错可能需要先将yolov3的模型先转换为416或者其他大小的模型,转换方法这里有:https://blog.csdn.net/qq_36780295/article/details/108496746

如果还是报错,可以直接按照这篇博客来,因为毕竟这个是已经成功了的。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值