基于特征知识蒸馏的人体姿态估计

• 作者: 袁泽昊 赵嘉莹
• 来源期刊: 软件
• 学科: 工学
• 关键词: 高分辨率网络 知识蒸馏 人体姿态估计 教师网络 学生网络
• 年，卷（期）: 2020,（12）
• 所属期刊栏目: 设计研究与应用
• 页码范围: 198-207
• 页数: 10页
• 分类号: TP391.41
• 语种: 中文
• DOI: 10.3969/j.issn.1003-6970.2020.12.046

摘要

现有的姿态估计只考虑如何提高模型的识别准确率,却忽略模型结构复杂和计算参数较多而无法在资源受限设备上高效运行的问题.针对这一问题,提出一种基于特征知识蒸馏的人体姿态估计模型.首先构建一种轻量级的高分辨率网络用于学习姿态信息.其次,提出一种新的知识蒸馏方法——特征蒸馏,与姿态蒸馏相结合,分别提取特征图中的高级语义信息和教师网络输出中的"软"知识,从不同角度指导学生网络的训练.实验结果表明,利用该方法得到的学生网络在MPII和COCO数据集上相比知识蒸馏所得到的学生网络识别准确率分别提升了0.6％和1.0％,计算成本却没有明显的增加.表明该方法能够有效的提高识别准确率,具有较高的成本效益.