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

评论 25
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值