Millimeter-wave radar offers a privacy-preserving and lighting-invariant alternative to RGB sensors for Human Pose Estimation (HPE) task. However, the radar signals are often sparse due to specular reflection, making the extraction of robust features from radar signals highly challenging. To address this, we present milliMamba, a radar-based 2D human pose estimation framework that jointly models spatio-temporal dependencies across both the feature extraction and decoding stages. Specifically, given the high dimensionality of radar inputs, we adopt a Cross-View Fusion Mamba encoder to efficiently extract spatio-temporal features from longer sequences with linear complexity. A Spatio-Temporal-Cross Attention decoder then predicts joint coordinates across multiple frames. Together, this spatio-temporal modeling pipeline enables the model to leverage contextual cues from neighboring frames and joints to infer missing joints caused by specular reflections. To reinforce motion smoothness, we incorporate a velocity loss alongside the standard keypoint loss during training. Experiments on the TransHuPR and HuPR datasets demonstrate that our method achieves significant performance improvements, exceeding the baselines by 11.0 AP and 14.6 AP, respectively, while maintaining reasonable complexity. Code: https://github.com/NYCU-MAPL/milliMamba


翻译:毫米波雷达为人体姿态估计任务提供了一种保护隐私且不受光照影响的替代方案。然而,由于镜面反射,雷达信号通常较为稀疏,这使得从雷达信号中提取鲁棒特征极具挑战性。为解决此问题,我们提出了milliMamba,一个基于雷达的二维人体姿态估计框架,该框架在特征提取和解码阶段联合建模时空依赖关系。具体而言,针对雷达输入的高维特性,我们采用了一种交叉视图融合Mamba编码器,以线性复杂度从更长的序列中高效提取时空特征。随后,一个时空交叉注意力解码器预测多帧中的关节坐标。这一时空建模流程共同使模型能够利用来自相邻帧和关节的上下文线索,来推断因镜面反射而缺失的关节。为了增强运动平滑性,我们在训练过程中除了标准的关键点损失外,还引入了速度损失。在TransHuPR和HuPR数据集上的实验表明,我们的方法取得了显著的性能提升,分别超过基线11.0 AP和14.6 AP,同时保持了合理的复杂度。代码:https://github.com/NYCU-MAPL/milliMamba

0
下载
关闭预览

相关内容

【CVPR2023】DynamicDet:目标检测的统一动态架构
专知会员服务
26+阅读 · 2023年4月15日
AAAI 2022 | ProtGNN:自解释图神经网络
专知
10+阅读 · 2022年2月28日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
46+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
VIP会员
相关基金
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
46+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
Top
微信扫码咨询专知VIP会员