Home

Awesome

JCDM

Joint Conditional Diffusion Model for Image Restoration with Mixed Degradations

method

Train and test

training

For the first stage training, modify the path of the dataset in config/diffusion.json.

python train_stage1.py

The trained models will be placed in the experiments/checkpoint folder.

testing

For the first stage testing, modify the path of the dataset and the pretrained model in config/infer.json.

python test_stage1.py