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