基于知识蒸馏的轻量级人体姿态估计网络设计

• 作者: 方芹 缪宁杰 董仲星 郑树松 王佳敏 罗文东 周霖
• 来源期刊: 机械设计与制造工程
• 学科: 工学
• 关键词: 姿态估计 知识蒸馏 教师网络 学生网络
• 年，卷（期）: 2021,（12）
• 所属期刊栏目: 机电一体化设计技术|Electromechanical integration design technology
• 页码范围: 64-68
• 页数: 5页
• 分类号: TM425
• 语种: 中文
• DOI: 10.3969/j.issn.2095-509X.2021.12.014

摘要

随着卷积神经网络迅速发展,人们通过设计复杂的模型来解决姿态估计问题.为提高运行效率,提出利用知识蒸馏将大型网络的知识转移到紧凑型网络中.首先训练一个深层次的姿态估计教师网络,然后利用教师网络预测得到的概率热图作为学生网络的标签,指导学生网络完成姿态估计.学生网络是一个轻量级的姿态估计网络,它可以学习教师网络的姿态结构知识,并且训练时间远低于教师网络,能够在计算资源不足的情况下训练学生网络.