Skeleton-GAN
A GCN-based generative adversarial network for skeleton sequence generation
Dependencies
- Python >= 3.6
- PyTorch >= 1.2.0
- pyyaml, tqdm, tensorboardX
Train
Dataset: NTU-RGB+D 120
Data preparation can be refered to: https://github.com/kenziyuliu/MS-G3D
python train.py --config ./config/train_cfg.yaml
Inference
python inference.py --config ./config/test_cfg.yaml --T 300 --B 10