HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Skeleton-based Action Recognition via Spatial and Temporal Transformer Networks

Chiara Plizzari Marco Cannici Matteo Matteucci

Skeleton-based Action Recognition via Spatial and Temporal Transformer Networks

Abstract

Skeleton-based Human Activity Recognition has achieved great interest in recent years as skeleton data has demonstrated being robust to illumination changes, body scales, dynamic camera views, and complex background. In particular, Spatial-Temporal Graph Convolutional Networks (ST-GCN) demonstrated to be effective in learning both spatial and temporal dependencies on non-Euclidean data such as skeleton graphs. Nevertheless, an effective encoding of the latent information underlying the 3D skeleton is still an open problem, especially when it comes to extracting effective information from joint motion patterns and their correlations. In this work, we propose a novel Spatial-Temporal Transformer network (ST-TR) which models dependencies between joints using the Transformer self-attention operator. In our ST-TR model, a Spatial Self-Attention module (SSA) is used to understand intra-frame interactions between different body parts, and a Temporal Self-Attention module (TSA) to model inter-frame correlations. The two are combined in a two-stream network, whose performance is evaluated on three large-scale datasets, NTU-RGB+D 60, NTU-RGB+D 120, and Kinetics Skeleton 400, consistently improving backbone results. Compared with methods that use the same input data, the proposed ST-TR achieves state-of-the-art performance on all datasets when using joints' coordinates as input, and results on-par with state-of-the-art when adding bones information.

Code Repositories

Chiaraplizz/ST-TR
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
skeleton-based-action-recognition-on-kineticsST-TR-agcn
Accuracy: 37.4
skeleton-based-action-recognition-on-ntu-rgbdST-TR-agcn
Accuracy (CS): 89.9
Accuracy (CV): 96.1
skeleton-based-action-recognition-on-ntu-rgbd-1ST-TR-agcn
Accuracy (Cross-Setup): 84.7%
Accuracy (Cross-Subject): 82.7%

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp