首页 > 其他分享 >【已解决】onnx转TensorRT遇到Assertion failed: axis >= 0 && axis < nbDims的问题

【已解决】onnx转TensorRT遇到Assertion failed: axis >= 0 && axis < nbDims的问题

时间:2024-03-27 18:29:31浏览次数:34  
标签:结点 nano onnx TensorRT Assertion 楼主 axis

最近在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

相关文章

  • ONNX 的简介及应用
    文章目录一、ONNX是什么二、ONNX具备哪些功能三、ONNX的相关概念四、ONNX的应用领域一、ONNX是什么ONNX(OpenNeuralNetworkExchange)是一种开放的深度学习模型交换格式,类似于JSON格式。它允许将模型从一个深度学习框架转换到另一个框架,以便在不同的平台和设备......
  • YOLOv8图像分割:使用ONNX模型进行推理
    基于COCO数据集的YOLOv8目标分割onnx模型推理在本博客中,我们将探讨如何使用YOLOv8目标分割模型进行推理,包括图片,视频文件,摄像头实时分割,特别是ONNX在不同大小(YOLOv8n,YOLOv8s,YOLOv8m,YOLOv8l,YOLOv8x)的模型上进行的实验。我们还将讨论所需的环境配置,代码实现,以及如何......
  • 【模型转换】PyTorch模型转ONNX模型
    我们主要使用torch.onnx.export()函数来实现PyTorch模型到ONNX模型的转换。importtorchimporttorchvision.modelsasmodelsdevice="cpu"#加载预训练的ResNet18模型model=models.resnet18(pretrained=True)model.eval().to(device)#定义输入input=torc......
  • C# Onnx C2PNet 图像去雾 室外场景
    目录介绍效果模型信息项目代码下载C#OnnxC2PNet图像去雾室外场景介绍github地址:https://github.com/YuZheng9/C2PNet[CVPR2023]CurricularContrastiveRegularizationforPhysics-awareSingleImageDehazing效果模型信息ModelProperties-----------......
  • C# danbooru Stable Diffusion 提示词反推 Onnx Demo
    目录说明效果模型信息项目代码下载 C#danbooruStableDiffusion提示词反推OnnxDemo说明模型下载地址:https://huggingface.co/deepghs/ml-danbooru-onnx效果模型信息Model Properties-----------------------------------------------------------------......
  • C# RAM Stable Diffusion 提示词反推 Onnx Demo
    目录介绍效果模型信息项目代码下载C#RAMStableDiffusion提示词反推OnnxDemo介绍github地址:https://github.com/xinyu1205/recognize-anythingOpen-sourceandstrongfoundationimagerecognitionmodels.onnx模型下载地址:https://huggingface.co/CannotFi......
  • onnx runtime文档学习5-加速Pytorch之Pytorch推理
    网上充斥着ONNXRuntime的简单科普,却没有一个系统介绍ONNXRuntime的博客,因此本博客旨在基于官方文档进行翻译与进一步的解释。ONNXruntime的官方文档:https://onnxruntime.ai/docs/如果尚不熟悉ONNX格式,可以参照该博客专栏,本专栏对onnx1.16文档进行翻译与进一步解释,ONNX......
  • onnx导出BERT示例
    onnx导出BERT示例目录onnx导出BERT示例BERT模型导出加载模型pt模型推理重新导出config导出onnx模型加载onnx测试参考资料BERT模型导出1.基于transformers载入PyTorch模型2.创建伪输入(dummyinputs),并利用伪输入在模型中前向inference,推理网络并在这个过程中追踪记录操作集合......
  • 无线表格识别模型LORE转换库:ConvertLOREToONNX
    引言总有小伙伴问到阿里的无线表格识别模型是如何转换为ONNX格式的。这个说来有些惭愧,现有的ONNX模型是很久之前转换的了,转换环境已经丢失,且没有做任何笔记。今天下定决心再次尝试转换,庆幸的是转换成功了。于是有了转换笔记:ConvertLOREToONNX。这次吸取教训,环境文件采用Anacond......
  • paddlehub下载时onnx下载失败
    0前言我下载paddlehub的时候(pipinstallpaddlehub)遇到了这个问题。实际上pipinstallonnx=1.8.1的时候也会有这个问题。note:Thiserrororiginatesfromasubprocess,andislikelynotaproblemwithpipERROR:FailedbuildingwheelforonnxailedtobuildonnxIR......