国内刊号:51-1307/TP
国际刊号:1001-9081
发布日期:
作者:李坤, 侯庆
单位:1.贵州大学 计算机科学与技术学院,贵阳 550025;2.贵州省通信产业服务有限公司,贵阳 550005
关键词:人体姿态估计,深度神经网络,高分辨率网络,深度可分离卷积,注意力机制
基金:国家创新型城市“百城百园”行动项目(筑科项目[2020]22号);贵州省大数据专项计划项目(20200424)
针对高分辨率人体姿态估计网络存在参数量大、运算复杂度高等问题,提出一种基于高分辨率网络(HRNet)的轻量型沙漏坐标注意力网络(SCANet)用于人体姿态估计。首先引入沙漏(Sandglass)模块和坐标注意力(CoordAttention)模块;然后在此基础上构建了沙漏坐标注意力瓶颈(SCAneck)模块和沙漏坐标注意力基础 (SCAblock)模块两种轻量型模块,在降低模型参数量和运算复杂度的同时,获取特征图空间方向的长程依赖和精确位置信息。实验结果显示,在相同图像分辨率和环境配置的情况下,在COCO(Common Objects in COntext)校验集上,SCANet模型与HRNet模型相比参数量降低了52.6%,运算复杂度降低了60.6%;在MPII(Max Planck Institute for Informatics)校验集上,SCANet模型与HRNet模型相比参数量和运算复杂度分别降低了52.6%和61.1%;与常见的人体姿态估计网络如堆叠沙漏网络(Hourglass)、级联金字塔网络(CPN)和SimpleBaseline相比,SCANet模型在拥有更少的参数量与运算复杂度的情况下,仍能实现对人体关键点的高准确度预测。
来源:2022年第8期
《计算机应用》期刊编辑部