11 lines (8 loc) · 592 Bytes

DiffusionDefogging

Diffusion Probabilistic Models for Image Defogging

dataset.py: Data preprocesing (Images downsampled to 512 x 256) and Dataloader functions
main.py: Training script
eval.py Evaluation metrics
run_diffusion.py Run the pre-trained model for evaluation and general inference.

Download the pre-trained model from here