Skip to content

Latest commit

 

History

History
49 lines (32 loc) · 918 Bytes

README.md

File metadata and controls

49 lines (32 loc) · 918 Bytes

event-diffusers

To run an extremely minimalistic implementation of DDPM on Event-Based Data (DVS-Gesture):

Install the requirements:

pip install -r requirements.txt

Or, using poetry:

poetry install

If you need to install poetry, make sure you are using a python version >= 3.10 and run:

curl -sSL https://install.python-poetry.org | python -

Add poetry to your path:

echo "export PATH=$PATH:$HOME/.local/bin" >> $HOME/.bashrc

Or if you're using zsh:

echo "export PATH=$PATH:$HOME/.local/bin" >> $HOME/.zshrc

To train the model:

python scripts/train_uncond_ddpm.py

For finetuning based on a pretrained Unconditional Imagenet Diffusion Model:

scripts/install_diffusion_pretrained.sh

To train on DVS-Gesture datasets:

IN PROGRESS