0


yolov8-pose:在yolov8上添加人体关键点检测

最近因为工作关系接触了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 所有, 如有侵权,请联系我们删除。

“yolov8-pose:在yolov8上添加人体关键点检测”的评论:

还没有评论