yolov9 tensorrt推理结果全为None
基于上述问题,我尝试以下几种方式:(1) 使用开源的基于yolov9的c++版推理代码,进行测试,发现结果认为nan值;(4) 最后我发现在使用export.py转onnx和engine时,因为要转为fp6格式,故参数里设置half为True,它会先将加载的pt模型截断到fp16范围内,再进行后续操作。在写部署代码时,发现使用python写推理代码,无论是fp32还是fp16的engine,最后结果都可以正常输出,但是在使用c++写同样的推理代码时发现使用fp32的engine作为引擎,推理结果正常。
作者:高的好想出去玩啊2024-04-24 22:32:53