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

decode_yolov8_pose.cu中的一点问题

Open scteam1994 opened this issue 1 year ago • 1 comments

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给覆盖了吗?

scteam1994 avatar Jan 17 '24 09:01 scteam1994

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

FeiYull avatar Feb 01 '24 00:02 FeiYull