arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.05421cs.CVcs.AI

EPAM-Net: 一种用于视频动作识别的高效姿态驱动注意力引导多模态网络

EPAM-Net: An Efficient Pose-driven Attention-guided Multimodal Network for Video Action Recognition

University of Louisville(路易斯维尔大学)

Ahmed Abdelkawy, Asem Ali, Aly Farag

AI总结:

针对现有多模态动作识别方法计算量大的问题,提出EPAM-Net,通过X-ShiftNet和时空注意力模块实现高效学习,在多个数据集上以更低计算量取得优异性能。

摘要:

现有的基于多模态的人体动作识别方法计算量巨大,限制了其在实时应用中的部署。在本研究中,我们提出了一种新颖且高效的姿态驱动注意力引导多模态网络(EPAM-Net),用于视频中的动作识别。具体而言,我们为RGB流和姿态流设计了扩展时序移位(X-ShiftNet)卷积架构,以从RGB视频及其骨架序列中捕获时空特征。X-ShiftNet通过将时序移位模块(Temporal Shift Module, TSM)集成到高效的2D CNN中,解决了3D CNN计算成本高的问题,实现了高效的时空学习。随后,我们利用骨架特征引导视觉网络流,通过提出的时空注意力模块聚焦关键帧及其显著空间区域。最后,融合两个流的预测结果以得到最终分类。实验结果表明,我们的方法在浮点运算量(FLOPs)大幅降低的情况下,在NTU RGB-D 60、NTU RGB-D 120、PKU-MMD和Toyota SmartHome数据集上的性能优于或可与当前最优方法相媲美。所提出的EPAM-Net的FLOPs最多降低了72.8倍,网络参数量最多降低了48.6倍。代码将在https://github.com/ahmed-nady/Multimodal-Action-Recognition上提供。

英文摘要:

Existing multimodal-based human action recognition approaches are computationally intensive, limiting their deployment in real-time applications. In this work, we present a novel and efficient pose-driven attention-guided multimodal network (EPAM-Net) for action recognition in videos. Specifically, we propose eXpand temporal Shift (X-ShiftNet) convolutional architectures for RGB and pose streams to capture spatio-temporal features from RGB videos and their skeleton sequences. The X-ShiftNet tackles the high computational cost of the 3D CNNs by integrating the Temporal Shift Module (TSM) into an efficient 2D CNN, enabling efficient spatiotemporal learning. Then skeleton features are utilized to guide the visual network stream, focusing on keyframes and their salient spatial regions using the proposed spatial-temporal attention block. Finally, the predictions of the two streams are fused for final classification. The experimental results show that our method, with a significant reduction in floating-point operations (FLOPs), outperforms and competes with the state-of-the-art methods on NTU RGB-D 60, NTU RGB-D 120, PKU-MMD, and Toyota SmartHome datasets. The proposed EPAM-Net provides up to a 72.8x reduction in FLOPs and up to a 48.6x reduction in the number of network parameters. The code will be available at https://github.com/ahmed-nady/Multimodal-Action-Recognition.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑