MEGADance

https://github.com/xulongt/megadance
Admin
MEGADance is a novel architecture for music-driven 3D dance generation that addresses challenges in...
Tokens:792
Snippets:16
Trust Score:6.5
Update:3 months ago
Show doc for...
Tokens:
Raw
### Install Remaining Dependencies

Source: https://github.com/xulongt/megadance/blob/main/README.md

Install all other project dependencies listed in the requirements.txt file.

```bash
pip install -r requirements.txt
```

--------------------------------

### Create Conda Environment

Source: https://github.com/xulongt/megadance/blob/main/README.md

Use this command to create a new conda environment for the project. Ensure Python 3.10 is installed.

```bash
conda create -n MEGA_env python=3.10
conda activate MEGA_env
```

--------------------------------

### Install PyTorch with CUDA 12.8

Source: https://github.com/xulongt/megadance/blob/main/README.md

Install the required PyTorch version with CUDA 12.8 support. This command uses the official PyTorch wheel index.

```bash
pip install torch==2.7.1+cu128 torchvision==0.22.1+cu128 torchaudio==2.7.1+cu128 \
    --index-url https://download.pytorch.org/whl/cu128
```

--------------------------------

### Train Stage 1 Model (FSQ)

Source: https://github.com/xulongt/megadance/blob/main/README.md

Run this command to train the Stage 1 model, which involves Finite Scalar Quantization (FSQ).

```bash
python train_fsq.py
```

--------------------------------

### Evaluate Stage 2 Model

Source: https://github.com/xulongt/megadance/blob/main/README.md

Run this command to evaluate the performance of the Stage 2 model.

```bash
python test_gpt.py
```

--------------------------------

### Evaluate Stage 1 Model

Source: https://github.com/xulongt/megadance/blob/main/README.md

Use this command to evaluate the performance of the Stage 1 model.

```bash
python test_fsq.py
```

--------------------------------

### Generate Dance Motion from Music

Source: https://github.com/xulongt/megadance/blob/main/README.md

This command initiates the inference process to generate dance motion based on a given music clip. Specify the root directory for demo files.

```bash
python demo_gpt.py --root_dir ./demo/1
```

--------------------------------

### Train Stage 2 Model (GPT)

Source: https://github.com/xulongt/megadance/blob/main/README.md

Execute this command to train the Stage 2 model, which is the Genre-Aware Dance Generation Stage (GADG).

```bash
python train_gpt.py
```

--------------------------------

### Train Dance Genre Classifier

Source: https://github.com/xulongt/megadance/blob/main/README.md

Execute this command to train the dance genre classifier model.

```bash
python train_cls.py
```
Tokens:
Raw