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

yolov8模型换成姿态模型后,怎么获取关键点列表 #36

Open zhaork526 opened 1 year ago

zhaork526 commented 1 year ago

global void decode_yolov8_device_kernel 函数最后添加下添加

for (size_t i = 0; i < 51; i++){ pout_item++ = (pitem+1+i); } 是否正确?