Home

Awesome

BlindDiff

Introduction

This is the official code of our work BlindDiff: Empowering Degradation Modelling in Diffusion Models for Blind Image Super-Resolution.

The pretrained model is here.

This repo is built on the basis of BasicSR and guided-diffusion, thanks for their open-sourcing!

Environment

Installations

Run the command:

pip install -r requirement.txt

and

python setup.py develop

Train

  1. Download trainning dataset DIV2K and Flickr2K for the natural images and FFHQ for the face images.
  2. Configure options/train.yml for your training.
  3. Run the command:
python basicsr/train.py -opt=options/train_setting.yml

Test

  1. Configure options/test.yml for your training. The testing dataset used in the paper is here.
  2. Run the command:
python basicsr/train.py -opt=options/test.yml