融合注意力机制的人体关键姿态估计
Human key pose estimation based on attention mechanism作者机构:中国民航大学计算机科学与技术学院天津300300 中国民用航空局智慧机场理论与系统重点实验室天津300300
出 版 物:《计算机工程与设计》 (Computer Engineering and Design)
年 卷 期:2023年第44卷第9期
页 面:2754-2760页
学科分类:08[工学] 081203[工学-计算机应用技术] 0812[工学-计算机科学与技术(可授工学、理学学位)]
基 金:天津市教委科研计划基金项目(2021KJ037) 中央高校基本科研业务费基金项目(3122021052) 民航航空公司人工智能重点实验室自主课题基金项目(CZAILAB-COO-KJAI20001)。
主 题:人体姿态估计 目标关键姿态 通道信息损失 特征冗余 空间注意力 通道注意力 代价敏感
摘 要:为提高与特定动作相关的目标关键姿态识别准确率,提出融合注意力机制的人体关键姿态估计方法。针对特征提取冗余和通道信息损失问题,在网络不同阶段融入注意力机制,帮助模型从空间和通道维度提取对任务更加重要的特征;考虑与特定动作相关的目标关键姿态误识别代价更大,设计一种代价敏感损失函数,提升模型对目标关键姿态的识别性能。实验结果表明,所提方法在COCO数据集和CargoSorting数据集上对目标关键姿态识别精度分别提高了1.4%和1.5%。