Kebii / Zoom-Transformer

Zoom Transformer for Skeleton-based Group Activity Recognition

Zoom-Transformer

Dependencies

• Python >= 3.6
• PyTorch >= 1.2.0
• pyyaml, tqdm, tensorboardX

Train

Dataset: Volleyball-Skeleton-Activity

Download: VS-A

python main.py --config ./config/train_cfg.yaml

Inference

python main.py --config ./config/test_cfg.yaml