This repository is built in PyTorch 1.13.1 and Python 3.8.11. Follow these intructions
- Clone our repository
git clone https://github.com/c-yn/AdaIR.git
cd AdaIR
- Create conda environment The Conda environment used can be recreated using the env.yml file
conda env create -f env.yml
All the datasets for 5 tasks used in the paper can be downloaded from the following locations:
Denoising: BSD400, WED, Urban100, Kodak24, BSD68
Deraining: Train100L&Rain100L
Dehazing: Train RESIDE, Test SOTS-Outdoor
Deblur: GoPro
Enhance: LOL-V1
The training data should be placed in data/Train/{task_name} directory where task_name can be Denoise, Derain, Dehaze, Deblur, or Enhance.
The testing data should be placed in the test directory wherein each task has a seperate directory.
The training and test datasets are organized as:
|--data
| |--Train
| | |--Deblur
| | | |--blur
| | | | |--GOPR0372_07_00_000047.png
| | | | |--GOPR0372_07_00_000048.png
| | | | ...
| | | |--sharp
| | | | |--GOPR0372_07_00_000047.png
| | | | |--GOPR0372_07_00_000048.png
| | | | ...
| | |--Dehaze
| | | |--original
| | | | |--0025.png
| | | | |--0039.png
| | | | ...
| | | |--synthetic
| | | | |--part1
| | | | | |--0025_0.8_0.1.jpg
| | | | | |--0025_0.8_0.2.jpg
| | | | | ...
| | | | |--part2
| | | | | |--3068_0.8_0.1.jpg
| | | | | |--3068_0.8_0.2.jpg
| | | | | ...
| | | | |--part3
| | | | | |--5425_0.8_0.1.jpg
| | | | | |--5425_0.8_0.2.jpg
| | | | | ...
| | | | |--part4
| | | | | |--6823_0.8_0.1.jpg
| | | | | |--6823_0.8_0.2.jpg
| | | | | ...
| | |--Denoise
| | | |--00001.bmp
| | | |--00001.bmp
| | | ...
| | | |--5096.jpg
| | | |--6046.jpg
| | | ...
| | |--Derain
| | | |--gt
| | | | |--norain-1.png
| | | | |--norain-2.png
| | | | ...
| | | |--rainy
| | | | |--rain-1.png
| | | | |--rain-2.png
| | | | ...
| | |--Enhance
| | | |--gt
| | | | |--2.png
| | | | |--5.png
| | | | ...
| | | |--low
| | | | |--2.png
| | | | |--5.png
| | | | ...
| |--test
| | |--deblur
| | | |--gopro
| | | | |--input
| | | | | |--GOPR0384_11_00_000001.png
| | | | | |--GOPR0384_11_00_000002.png
| | | | | ...
| | | | |--target
| | | | | |--GOPR0384_11_00_000001.png
| | | | | |--GOPR0384_11_00_000002.png
| | | | | ...
| | |--dehaze
| | | |--input
| | | | |--0001_0.8_0.2.jpg
| | | | |--0002_0.8_0.08.jpg
| | | | ...
| | | |--target
| | | | |--0001.png
| | | | |--0002.png
| | | | ...
| | |--denoise
| | | |--bsd68
| | | | |--3096.jpg
| | | | |--12084.jpg
| | | | ...
| | | |--urban100
| | | | |--img_001.png
| | | | |--img_002.png
| | | | ...
| | | |--kodak24
| | | | |--kodim01.png
| | | | |--kodim02.png
| | | | ...
| | |--derain
| | | |--Rain100L
| | | | |--input
| | | | | |--1.png
| | | | | |--2.png
| | | | | ...
| | | | |--target
| | | | | |--1.png
| | | | | |--2.png
| | | | | ...
| | |--enhance
| | | |--lol
| | | | |--input
| | | | | |--1.png
| | | | | |--22.png
| | | | | ...
| | | | |--target
| | | | | |--1.png
| | | | | |--22.png
| | | | | ...