yolov9 tensorrt推理结果全为None

devtools/2024/12/22 19:56:14/

  使用yolov9源码自带export.py函数将pt转为onnx和engine。在写部署代码时,发现使用python写推理代码,无论是fp32还是fp16的engine,最后结果都可以正常输出,但是在使用c++写同样的推理代码时发现使用fp32的engine作为引擎,推理结果正常。而使用fp16的engine作为推理引擎时,输出结果全为nan和0值。已知的是TensorRT版本一致。c++上fp32的engine正常输出,说明代码中预处理没问题,问题肯定就出在转出的engine上。

  基于上述问题,我尝试以下几种方式:(1) 使用开源的基于yolov9的c++版推理代码,进行测试,发现结果认为nan值;(2) 使用预训练模型转为fp16-engine进行推理,发现结果仍为nan;(3) 尝试换用trtexec命令直接转fp16-engine,发现结果认为nan;(4) 最后我发现在使用export.py转onnx和engine时,因为要转为fp6格式,故参数里设置half为True,它会先将加载的pt模型截断到fp16范围内,再进行后续操作。这让我想起之前在转其他模型时这个地方同样出过错,故这里使用export.py转onnx,并将half设为False。转为onnx后再使用trtexec转为fp16范围内的engine。再次推理发现,结果正常输出。
  问题是解决了,但是我目前还不知道为什么,有知道的友友吗?


http://www.ppmy.cn/devtools/13623.html

相关文章

SecureCRT (Mac/Windows)中文---远程连接与管理的安全新选择

SecureCRT是一款功能强大的终端仿真程序,专为连接和管理远程系统而设计。它支持SSH(包括SSH1和SSH2)协议,确保用户与目标设备之间的通信安全,有效防止网络攻击和窥探。此外,SecureCRT还支持Telnet、Rlogin等…

2024最新SSL证书在线申请系统源码 | 支持API接口 支持在线付费 二开优化版

内容目录 一、详细介绍二、效果展示1.部分代码2.效果图展示 三、学习资料下载 一、详细介绍 2024最新SSL证书在线申请系统源码 | 支持API接口 支持在线付费 二开优化版 最新SSL证书在线申请系统源码 | 支持API接口 SSL证书保证网络安全的基本保障。向您介绍我们的在线生成SSL…

2. uni-app的一些介绍

前言 就目前的前端生态而言,跨端开发基本算是每一个前端开发者必备的技能点之一了,而在Vue这个技术栈里uni-app在跨端是独一档的,不信的话可以翻翻Boss之类的招聘网站.... 概述 阅读时间:约5~10分钟; 本文重点&…

容器云平台运维的范围与架构设计新思潮

容器云平台运维的范围与架构设计 【导读】容器云平台有其独特的特点,不同于传统系统的运维。本文分享了作者对容器云平台运维范围和运维架构设计的思考与实践。 一、容器云平台运维范围 (一) 梳理要运维哪些内容 作为运维专家,…

nn.Sequential与tensorflow的Sequential对比

nn.Sequential() 是 PyTorch 深度学习框架中的一个类,用于按顺序容器化模块。nn.Sequential 是一个有序的容器,它包含多个网络层,数据会按照在构造函数中传入顺序依次通过每个层。在 nn.Sequential 中,不需要定义 forward 方法&am…

Maven:配置与使用指南1

https://mvnrepository.com Maven 1.maven简介 不同模块的jar包以及同时设计的功能的微小变化版本; 真实的开发环境:我们将我们的源代码在服务器上重新编译重新打包,工程升级维护过程繁琐 1.Maven是一个项目管理工具,将项目开…

基于Python的图书借阅管理系统,附源码

博主介绍:✌IT徐师兄、7年大厂程序员经历。全网粉丝15W、csdn博客专家、掘金/华为云//InfoQ等平台优质作者、专注于Java技术领域和毕业项目实战✌ 🍅文末获取源码联系🍅 👇🏻 精彩专栏推荐订阅👇&#x1f3…

wordpress建网站主题案例推荐

wordpress企业网站主题案例 https://www.mymoban.com/wordpress/ wordpress公司官网主题案例 https://www.wowsoho.com/jianzhan wordpress外贸主题案例 https://www.wpniu.com/moban