最近在jetson nano上部署YOLOv8的时候遇到了许多问题(参考jetson nano 部署YOLOv8),大部分比较好解决,其中在将模型部署到jetson nano的阶段遇到了如下两个问题:
静态onnx转TensorRT遇到Assertion failed: axis >= 0 && axis < nbDims问题
动态onnx转tensorrt报错:Attribute not found: axes的问题
首先对模型进行了简化,但并没有解决问题。因为第一个问题困扰很久没有解决所以先尝试第二个,在博客和github上查找了很久,最后发现是生成的结点问题,大部分人给出的解法无非两种:
1、更改官方的代码,见这里。2、对模型结点进行修改,自己diy,见这里。
大部分看官通过这两种方法解决了,楼主先天非酋圣体没有解决,并且再使用第二种方法修改结点之后报错"Assertion failed: axis >= 0 && axis < nbDims"属实是闭环了
标签:结点,nano,onnx,TensorRT,Assertion,楼主,axis From: https://blog.csdn.net/weixin_75219261/article/details/137084620