AdaIR

KO | EN

gitlite — search

#python #hacktoberfest #react #typescript #llm #machine-learning #rust #linux #docker #nodejs #windows #awesome #kubernetes #mcp

(★ 257)

[ICLR'25] AdaIR: Adaptive All-in-One Image Restoration via Frequency Mining and Modulation

File Explorer

Download Latest Version (.zip)

- - hazy_outside.txt
- - denoise.txt
- - rainTrain.txt
- AdaIR.png
- adair3d.PNG
- adair5d.PNG
- model.py
- - __init__.py
- __init__.py
- dataset_utils.py
- degradation_utils.py
- image_io.py
- image_utils.py
- imresize.py
- loss_utils.py
- schedulers.py
- val_utils.py
.gitignore
env.yaml
INSTALL.md
LICENSE
options.py
README.md
test.py
train.py

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# AdaIR: Adaptive All-in-One Image Restoration via Frequency Mining and Modulation (ICLR'25)

Yuning Cui, [Syed Waqas Zamir](https://scholar.google.ae/citations?hl=en&user=POoai-QAAAAJ), [Salman Khan](https://salman-h-khan.github.io/), [Alois Knoll](https://scholar.google.com.hk/citations?user=-CA8QgwAAAAJ&hl=zh-CN&oi=ao), [Mubarak Shah](https://scholar.google.com.hk/citations?user=p8gsO3gAAAAJ&hl=zh-CN&oi=ao), and [Fahad Shahbaz Khan](https://scholar.google.es/citations?user=zvaeYnUAAAAJ&hl=en)

## 🎄 We have released the code and models for our new all-in-one model, which achieves higher performance with lower computational overhead.
## 🎁 Please refer to [BioIR](https://github.com/c-yn/BioIR) (NeurIPS 2025)

<hr />

> **Abstract:** *In the image acquisition process, various forms of degradation, including noise, blur, haze, and rain, are frequently introduced. These degradations typically arise from the inherent limitations of cameras or unfavorable ambient conditions. To recover clean images from their degraded versions, numerous specialized restoration methods have been developed, each targeting a specific type of degradation. Recently, all-in-one algorithms have garnered significant attention by addressing different types of degradations within a single model without requiring the prior information of the input degradation type. However, these methods purely operate in the spatial domain and do not delve into the distinct frequency variations inherent to different degradation types. To address this gap, we propose an adaptive all-in-one image restoration network based on frequency mining and modulation. Our approach is motivated by the observation that different degradation types impact the image content on different frequency subbands, thereby requiring different treatments for each restoration task. Specifically, we first mine low- and high-frequency information from the input features, guided by the adaptively decoupled spectra of the degraded image. The extracted features are then modulated by a bidirectional operator to facilitate interactions between different frequency components. Finally, the modulated features are merged into the original input for a progressively guided restoration. With this approach, the model achieves adaptive reconstruction by accentuating the informative frequency subbands according to different input degradations. Extensive experiments demonstrate that the proposed method, named AdaIR, achieves state-of-the-art performance on different image restoration tasks, including image denoising, dehazing, deraining, motion deblurring, and low-light image enhancement.* 
<hr />

## Network Architecture
<img src = "figs/AdaIR.png">

## Installation and Data Preparation

See [INSTALL.md](INSTALL.md) for the installation of dependencies and dataset preperation required to run this codebase.

## Training

After preparing the training data in ```data/``` directory, use 
```
python train.py
```
to start the training of the model. Use the ```de_type``` argument to choose the combination of degradation types to train on. By default it is set to all the 5 degradation tasks (denoising, deraining, dehazing, deblurring, enhancement).

Example Usage: If we only want to train on deraining and dehazing:
```
python train.py --de_type derain dehaze
```

## Testing

After preparing the testing data in ```test/``` directory, place the mode checkpoint file in the ```ckpt``` directory. The pre-trained model can be downloaded [here](https://drive.google.com/drive/folders/1x2LN4kWkO3S65jJlH-1INUFiYt8KFzPH?usp=sharing). To perform the evaluation, use
```
python test.py --mode {n}
```
```n``` is a number that can be used to set the tasks to be evaluated on, 0 for denoising, 1 for deraining, 2 for dehazing, 3 for deblurring, 4 for enhancement, 5 for three-degradation all-in-one setting and 6 for five-degradation all-in-one setting.

Example Usage: To test on all the degradation types at once, run:

```
python test.py --mode 6
```

## Results
Performance results of the AdaIR framework trained under the all-in-one setting.

<details>
<summary>Three Distinct Degradations (click to expand) </summary>

<img src = "figs/adair3d.PNG"> 
</details>
<details>
<summary>Five Distinct Degradations (click to expand) </summary>

The visual results can be downloaded [here](https://drive.google.com/drive/folders/1lsYFumrn3-07Vcl3TZy0dzMMA9yDTpSK?usp=sharing).

## Citation
If you use our work, please consider citing:
~~~
@inproceedings{cui2025adair,
title={Ada{IR}: Adaptive All-in-One Image Restoration via Frequency Mining and Modulation},
author={Yuning Cui and Syed Waqas Zamir and Salman Khan and Alois Knoll and Mubarak Shah and Fahad Shahbaz Khan},
booktitle={The Thirteenth International Conference on Learning Representations},
year={2025}
}
~~~

## Contact
Should you have any questions, please contact yuning.cui@in.tum.de

**Acknowledgment:** This code is based on the [PromptIR](https://github.com/va1shn9v/PromptIR) repository.

AdaIR

File Explorer

Today's picks

Was this content helpful?

Recent Feedback

AdaIR

File Explorer

Today's picks

Was this content helpful?

Recent Feedback

Download README