Skip to content

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

Notifications You must be signed in to change notification settings

p4r4mount/E-Motion

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

[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

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages