decode_yolov8_pose.cu中的一点问题

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

decode_yolov8_pose.cu中的一点问题 #73

Open scteam1994 opened 6 months ago

scteam1994 commented 6 months ago

https://github.com/FeiYull/TensorRT-Alpha/blob/403291186bb91fa69c76907b68773de94fa7c15a/yolov8-pose/decode_yolov8_pose.cu#L42

decode_yolov8_pose_device_kernel中最后一行让我有些疑惑。在41行pout_item++ = 1;按照逻辑应该是先让pout_item往前偏移一位，然后在赋值1，如果这样的话，在memcpy的时候是不是少了一个+ 1，应该memcpy(pout_item+ 1, pitem + 1, (dstWidth - 7) sizeof(float));否则不是把刚刚赋值的keepflag给覆盖了吗?

FeiYull commented 5 months ago

@scteam1994 40行判断是否是person，41行是判断是否 keep