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

yolo系列模型前处理都没有alpha beta norm吗? #71

Open scteam1994 opened 6 months ago

scteam1994 commented 6 months ago

我看到kernel里有norm device函数,但是默认值是0,1,也没有修改过,但是我看官方repo里v5是包含/255的前处理的,请问是把/255存入模型中了,还是怎么回事?

FeiYull commented 6 months ago

Kernel里有除255

scteam1994 commented 6 months ago

我找到了