最近因为工作关系接触了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以及是否可见文章来源:https://www.toymoban.com/news/detail-504431.html
有兴趣的朋友可以到我github查看完整代码,包含训练以及推理,https://github.com/yts1111/yolov8-pose文章来源地址https://www.toymoban.com/news/detail-504431.html
到了这里,关于yolov8-pose:在yolov8上添加人体关键点检测的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!