Slightly modify and debug from https://github.com/rosinality/denoising-diffusion-pytorch to allow easy colab training and inference
Imeplementation of https://arxiv.org/abs/2006.11239
Purpose: train a diffusion model to generate data for augmentation of Parkinson's disease patient's drawing
Use this following link to start training in colab https://colab.research.google.com/drive/1KKy9llE8GWEWCXYQPOoM1lm7hdpBYlrC?usp=sharing
Example generated image