README.md

NDR-Restore

[TIP24] Neural Degradation Representation Learning for All-In-One Image Restoration

Mingde Yao, Ruikang Xu, Yuanshen Guan, Jie Huang, Zhiwei Xiong

University of Science and Technology of China

[Paper] [BibTeX] ⚡ 🚀 🔥

🚀 Welcome! This is the official repository of Neural Degradation Representation Learning for All-In-One Image Restoration.

Training and testing

Modify the configurations in *.yml.

Run the demo.

Download the pretrained-model

python demo.py -opt options/demo.yml

Train

python train.py -opt options/train.yml

Test

python test.py -opt options/test.yml

❤️ Citing Us

If you find this repository or our work useful, please consider giving a star ⭐ and citation 🦖 , which would be greatly appreciated:

@article{yao2023neural,
  title={Neural Degradation Representation Learning for All-In-One Image Restoration},
  author={Yao, Mingde and Xu, Ruikang and Guan, Yuanshen and Huang, Jie and Xiong, Zhiwei},
  journal={arXiv preprint arXiv:2310.12848},
  year={2023}
}

📧 Contact

For any inquiries or questions, please contact me by email ([email protected]).