改进KAPAO的人体关键点检测

作者：赵普; 武一^*

来源：电子测量与仪器学报, 2023, 37(07): 177-185.

DOI：10.13382/j.jemi.B2306196

• 关键点检测
• 非极大值抑制
• 注意力特征融合
• 数据增强

针对人体关键点检测存在检测精确度低的不足，在KAPAO(keypoints and pose as objects)网络的基础上进行改进。使用PoseTrans(pose transformation)进行数据增强，提高网络的泛化性；针对特征融合能力的不足，设计融合注意力机制的BiFPN(Bi-directional feature network)模块充分融合不同语义特征，提高网络对深层语义信息和浅层语义信息的融合能力；在网络输出阶段设计自适应扩张卷积模块，将不同扩张率的输出分支进行自适应融合，有效获得图像的全局信息；在网络的后处理部分设计SDR-NMS(soft DIOU relocation non-maximum suppression)替代传统的NMS,保留最优的关键点预测框。实验结果表明，网络的AP分数提高了4.8%,AP为68.6%,检测速度为19.1 ms。网络精确度和检测速度均具有较好的表现性。

• 单位
  电子信息工程学院; 河北工业大学