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