This project employs the FineDance dataset for both training and evaluation.
-
FineDance Dataset
Download from Google Drive and extract to./data:
Google Drive Link -
SMPL Models
Download required SMPL models from the official SMPL-X website and place them in./assets:
SMPL/SMPL-X Official Site -
Textual Descriptive Dataset
(To be released soon)
- Motion feature extraction
python preprocess/pre_motion.py \
--motion_dir ./data/finedance/motion \
--store_dir ./data/train/motion_fea319- Music feature extraction
python preprocess/pre_music.py \
--music_dir ./data/finedance/music_wav \
--store_dir ./data/train/musicWe provide a pretrained GCDance checkpoint for evaluation and generation:
Download it and place it under ./checkpoint, then point to it with the --checkpoint argument when running test.py / test_10.py, e.g.:
python test.py --test_genTo train the model:
accelerate launch train.py --wandb We provide two multi-task optimization strategies. Note: Aligned training requires high GPU memory (recommended:2 NVIDIA A100).
Aligned Training
accelerate launch train.py --wandb --mtl_method AlignedNash Training
accelerate launch train.py --wandb --mtl_method NashDuring evaluation, 18 music tracks are segmented into multiple 120-frame clips. For each track, 15 clips are randomly selected (totaling 270 segments) and repeated 10 times to compute the mean performance.
To prepare the test data, you may either:
- Option A — Process the test data manually:
python slice_test_data.py- Option B — Use the preprocessed test data: Download from Google Drive and place it in the project root directory: Google Drive
After preparing the data, run the following command to evaluate the randomly selected 270 motion segments:
python test_10.pypython test.py --test_genpython vis.py --motion_save_dir If you find this work useful, please consider citing:
@article{liu2026gcdance,
title={GCDance: Genre-Controlled Music-Driven 3D Full Body Dance Generation},
author={Liu, Xinran and Dong, Xu and Qian, Shenbin and Kanojia, Diptesh and Wang, Wenwu and Feng, Zhenhua},
journal={IEEE Transactions on Multimedia},
year={2026},
publisher={IEEE}
}We would like to express our sincere gratitude to the EDGE and FineDance projects for their foundational datasets and research support.
