yolov5 rt推理结果和pytorch的推理结果得分差别很大

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎

GNU General Public License v2.0

1.26k stars 196 forks source link

yolov5 rt推理结果和pytorch的推理结果得分差别很大 #66

Open Waynepoo opened 6 months ago

Waynepoo commented 6 months ago

上图是pytorch yolov5s_v7.0 640640推理结果上图是yolov5s_v7.0 rt 640640推理结果为什么会出现这么大的偏差？

FeiYull commented 6 months ago

@Waynepoo
先说结论：不影响。

pytorch使用的是矩形推理（rectangle inference），将图像的HW缩放为32的倍数（H W中较大者等于640），然后直接推理。
为了高效推理，仓库中使用tensorrt推理，是强制将任何分辨率的图像缩放 + padding为640X640。

Waynepoo commented 6 months ago

@Waynepoo 先说结论：不影响。

pytorch使用的是矩形推理（rectangle inference），将图像的HW缩放为32的倍数（H W中较大者等于640），然后直接推理。

为了高效推理，仓库中使用tensorrt推理，是强制将任何分辨率的图像缩放 + padding为640X640。

谢谢，懂了