RDM-IR: Task-Adaptive Deep Unfolding Network for All-In-One Image Restoration(in submission)

Official PyTorch implementation of DRM-IR: Task-Adaptive Deep Unfolding Network for All-In-One Image Restoration.

The complete code will be open source in this repository.

Framework

Requirements

1. Python==3.6.13
2. Pytorch==1.10.1
3. torchvision==0.11.2

Train

Evaluate

Visualization of results

Citation

If you find our work useful, please cite our paper by the following:

@article{cheng2023drm,
  title={DRM-IR: Task-Adaptive Deep Unfolding Network for All-In-One Image Restoration},
  author={Cheng, Yuanshuo and Shao, Mingwen and Wan, Yecong and Wang, Chao and Zuo, Wangmeng},
  journal={arXiv preprint arXiv:2307.07688},
  year={2023}
}

Noted that the code is only for non-commercial use! Should you have any queries, contact me at cys1294414023@gmail.com