发明公开
- 专利标题: 改进YOLOv5的多人姿态估计修正算法
-
申请号: CN202311210876.4申请日: 2023-09-18
-
公开(公告)号: CN117351515A公开(公告)日: 2024-01-05
- 发明人: 吴思 , 贾迪 , 赵金源 , 张博 , 宁博 , 杜成名 , 左越 , 李浛永 , 郭奕男 , 罗玲 , 王健伟 , 刘佳明
- 申请人: 国网辽宁省电力有限公司葫芦岛供电公司 , 国网辽宁省电力有限公司 , 国家电网有限公司
- 申请人地址: 辽宁省葫芦岛市龙港区龙湾大街1号
- 专利权人: 国网辽宁省电力有限公司葫芦岛供电公司,国网辽宁省电力有限公司,国家电网有限公司
- 当前专利权人: 国网辽宁省电力有限公司葫芦岛供电公司,国网辽宁省电力有限公司,国家电网有限公司
- 当前专利权人地址: 辽宁省葫芦岛市龙港区龙湾大街1号
- 代理机构: 葫芦岛天开专利商标代理事务所
- 代理商 盖文双
- 主分类号: G06V40/10
- IPC分类号: G06V40/10 ; G06V10/25 ; G06V10/82 ; G06N3/0455 ; G06N3/0464 ; G06N3/082
摘要:
本发明提出了改进YOLOv5的多人姿态估计修正算法。该方法通过在YOLOv5的骨干网络中,融入跳跃注意力模块,再颈部网络中利用加权双向特征金字塔并联合使用跳跃注意力模块与Transformer编码器,在检测部分增加检测头,使网络对微小目标更加敏感,最后,利用网络预测得到关键点对象信息并对其修整,解决高密度人群中小尺度目标的姿态预测的技术问题。本发明提高网络对兴趣区域的特征提取能力,聚焦于有用的目标对象,加强网络特征融合能力,使网络更加关注全局信息;增加小目标检测检测头,使网络更加关注小尺度目标;最后结合关键点信息对姿态信息进行修正,最终预测结果较YOLOv5在COCO数据集上AP50提高2.2%,AP75提高3.3%。适宜作为高密度人群中小尺度目标的姿态预测精度方法应用。