基于注意力机制的轻量化人体姿态估计网络

作     者：黄凯伦 王峥 HUANG Kailun;WANG Zheng

作者机构：武汉邮电科学研究院湖北武汉430070 南京烽火天地通信科技有限公司江苏南京210019 

出 版 物：《电子设计工程》 (Electronic Design Engineering)

年 卷 期：2025年第33卷第2期

页      面：176-180页

学科分类：0810[工学-信息与通信工程] 08[工学] 081001[工学-通信与信息系统] 

主　　题：高分辨率网络 多头自注意力 混洗网络 深度可分离卷积 

摘      要：针对智能家居等移动设备计算资源和存储空间有限的问题,使用高分辨率网络作为主干,引入轻量级卷积神经网络ShuffleNetV2和注意力机制,提出了一种高效轻量的网络。用视觉Trans⁃former的多头自注意力层替换ShuffleNetV2的3×3深度可分离卷积,得到新模块。用新模块替换高分辨网络第二个3×3卷积及后三阶段的所有残差块。设计了一种新的输出模块,包含并行双注意力结构和先卷积后采样结构。通过设置不同的模块数量、通道数量和多头数量得到了两种大小的网络。实验结果表明,两种网络与其他主流人体姿态估计网络相比准确率更高,参数量下降50%以上,达到轻量化的目的。