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

    最近因为工作关系接触了yolo-pose,1月份yolov8也出来了,就想着能不能在yolov8上也加上pose分支,也算加深对网络的认识。

    yolov8在数据处理上也考虑了keypoints的,所以数据处理部分不用太多的修改,主要修改了Detect类、Loss类。

    Detect类:__init__方法中加入nkpt以及cv4分支,用于检测人体关键点

yolov8-pose:在yolov8上添加人体关键点检测_第1张图片

    Loss类:__call__方法中加入计算人体关键点的oks loss。

yolov8-pose:在yolov8上添加人体关键点检测_第2张图片 

yolov8-pose:在yolov8上添加人体关键点检测_第3张图片 

    修改完成后在我自己的小数据上能够收敛,训练数据格式如下:

    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

你可能感兴趣的:(深度学习,人工智能)