/E-Motion

PyTorch implementation of E-Motion: Future Motion Simulation via Event Sequence Diffusion

Primary LanguagePython

[NeurIPS 2024] E-Motion: Future Motion Simulation via Event Sequence Diffusion

Arvix: E-Motion: Future Motion Simulation via Event Sequence Diffusion.

🛠️ Requirements and Installation

  • Python >= 3.10
  • Pytorch == 2.0.1
  • CUDA Version >= 11.7
  • Install required packages:
git clone https://github.com/p4r4mount/E-Motion.git
cd E-Motion
conda env create -f environment.yml

💡 Pre-trained weights

Google Drive: Checkpoint

🚀 Inference

python predict.py --model_path /path/to/model/checkpoint \
                  --data_path /path/to/data/file.npy \
                  --output_path /path/to/output/directory