TensorRT-Alpha
TensorRT-Alpha copied to clipboard
yolov5 rt推理结果和pytorch的推理结果得分差别很大
上图是pytorch yolov5s_v7.0 640640推理结果
上图是yolov5s_v7.0 rt 640640推理结果
为什么会出现这么大的偏差?
@Waynepoo
先说结论:不影响。
-
pytorch使用的是矩形推理(rectangle inference),将图像的HW缩放为32的倍数(H W中较大者等于640),然后直接推理。
-
为了高效推理,仓库中使用tensorrt推理,是强制将任何分辨率的图像缩放 + padding为640X640。
@Waynepoo 先说结论:不影响。
- pytorch使用的是矩形推理(rectangle inference),将图像的HW缩放为32的倍数(H W中较大者等于640),然后直接推理。
- 为了高效推理,仓库中使用tensorrt推理,是强制将任何分辨率的图像缩放 + padding为640X640。
谢谢,懂了