YOLOv7_Tensorrt icon indicating copy to clipboard operation
YOLOv7_Tensorrt copied to clipboard

转换一切顺利,但是检测到的物体置信度为负数

Open Bin-ze opened this issue 2 years ago • 5 comments

我按照您的readme在开发板上进行了模型部署,一切顺利,但是检测物体的置信度为负数,这是因为什么原因造成的,因为我没有在trt推理下评估模型的指标,所以我很难确定这样带来的性能影响,如果你可以回答我的疑问,我将很感激

Bin-ze avatar Aug 21 '22 09:08 Bin-ze

因为tensorrt 8.2有bug。升级至8.4可以解决。

Monday-Leo avatar Aug 24 '22 17:08 Monday-Leo

这个bug为什么转换yolov5不出现,还有使用pip直接安装tensorrt不出现,只出现在自己源码安装tensorrt或许jston平台刷的tensorrt也有问题,感觉很奇怪

发自我的iPhone

在 2022年8月25日,01:10,Monday-Leo @.***> 写道:

 因为tensorrt 8.2有bug。升级至8.4可以解决。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.

Bin-ze avatar Aug 25 '22 00:08 Bin-ze

这是nms的bug,yolov5的nms没有并入到模型当中,所以没关系。

Monday-Leo avatar Aug 25 '22 10:08 Monday-Leo

那我自己写nms,倒出无nms的onnx模型呢

发自我的iPhone

在 2022年8月25日,18:35,Monday-Leo @.***> 写道:

 这是nms的bug,yolov5的nms没有并入到模型当中,所以没关系。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.

Bin-ze avatar Aug 25 '22 15:08 Bin-ze

那就没有问题,甚至可以用tensorrt7,这个仓库主要是添加了efficientnms到模型中,所以才使用的tensorrt8.4。

Monday-Leo avatar Aug 26 '22 07:08 Monday-Leo