hamadichihaoui / BIRD

This is the official implementation of "Blind Image Restoration via Fast Diffusion Inversion"
212 stars 18 forks source link

Blind Image Restoration via Fast Diffusion Inversion (BIRD)

This is the official implementation of "Blind Image Restoration via Fast Diffusion Inversion ". arxiv

Environment

pip install numpy torch blobfile tqdm pyYaml pillow diffusers   

Pre-Trained Models

For face restoration, download the pretrained model from and put it into checkpoints/.

Blind Deblurring

python blind_deblurring.py    

image info

Non-uniform Deblurring

python non_uniform_deblurring.py    

image info

Inpainting

python inpainting.py    

image info

Denoising

python denoising.py    

image info

Superresolution

python super_resolution.py    

image info

References

If you find this repository useful for your research, please cite the following work.

@article{chihaoui2024blind,
  title={Blind Image Restoration via Fast Diffusion Inversion},
  author={Chihaoui, Hamadi and Lemkhenter, Abdelhak and Favaro, Paolo},
  journal={arXiv preprint arXiv:2405.19572},
  year={2024}
}