TensorRT-Alpha icon indicating copy to clipboard operation
TensorRT-Alpha copied to clipboard

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

Open Waynepoo opened this issue 2 years ago • 2 comments

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

Waynepoo avatar Dec 27 '23 09:12 Waynepoo

@Waynepoo
先说结论:不影响。

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

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

FeiYull avatar Dec 27 '23 09:12 FeiYull

@Waynepoo 先说结论:不影响。

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

谢谢,懂了

Waynepoo avatar Dec 27 '23 09:12 Waynepoo