最近因为工作关系接触了yolo-pose，1月份yolov8也出来了，就想着能不能在yolov8上也加上pose分支，也算加深对网络的认识。
yolov8在数据处理上也考虑了keypoints的，所以数据处理部分不用太多的修改，主要修改了Detect类、Loss类。
Detect类：__init__方法中加入nkpt以及cv4分支，用于检测人体关键点
Loss类：__call__方法中加入计算人体关键点的oks loss。
修改完成后在我自己的小数据上能够收敛，训练数据格式如下：
0代表类别
0.71 0.534 0.228 0.44代表box的x，y，w，h
0.707 0.361 2代表该关键点的x，y以及是否可见
有兴趣的朋友可以到我github查看完整代码，包含训练以及推理，https://github.com/yts1111/yolov8-pose
本文转载自: https://blog.csdn.net/ytsaiztt/article/details/129876944
版权归原作者 ytsaiztt 所有， 如有侵权，请联系我们删除。
版权归原作者 ytsaiztt 所有， 如有侵权，请联系我们删除。