TensorRT-Alpha
TensorRT-Alpha copied to clipboard
decode_yolov8_pose.cu中的一点问题
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 40行判断是否是person,41行是判断是否 keep