├── .github └── workflows │ └── pylint.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .vscode └── settings.json ├── LICENSE ├── README.md ├── VERSION ├── datasets └── README.md ├── experiments └── README.md ├── options ├── test │ ├── 001_xrestormer_sr.yml │ ├── 002_xrestormer_denoise.yml │ ├── 003_xrestormer_deblur.yml │ ├── 004_xrestormer_derain.yml │ └── 005_xrestormer_dehaze.yml └── train │ ├── 001_xrestormer_s_sr.yml │ ├── 001_xrestormer_sr.yml │ ├── 002_xrestormer_denoise.yml │ ├── 003_xrestormer_deblur.yml │ ├── 004_xrestormer_derain.yml │ ├── 005_xrestormer_dehaze.yml │ ├── A11_MPRNet_SR_bs16p192.yml │ ├── A12_SwinIR_SR_bs16p192.yml │ ├── A13_Uformer_SR_bs16p192.yml │ ├── A14_Restomer_SR_bs16p192.yml │ └── A15_NAFNet_SR_bs16p192.yml ├── requirements.txt ├── scripts ├── prepare_example_data.py ├── test.sh └── train.sh ├── setup.cfg ├── setup.py └── xrestormer ├── __init__.py ├── archs ├── MPRNet_arch.py ├── NAFNet_arch.py ├── NAFSSR_arch.py ├── Uformer_arch.py ├── __init__.py ├── restormer_base_arch.py └── xrestormer_arch.py ├── data ├── __init__.py ├── dehaze_dataset.py ├── gaussian_noise_dataset.py └── size_paired_dataset.py ├── losses ├── __init__.py └── example_loss.py ├── models ├── MPRNet_model.py ├── __init__.py ├── lr_scheduler.py ├── restormer_model.py ├── uformer_model.py └── xrestormer_model.py ├── test.py ├── train.py └── version.py /.github/workflows/pylint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/.github/workflows/pylint.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 0.1.0 2 | -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/datasets/README.md -------------------------------------------------------------------------------- /experiments/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/experiments/README.md -------------------------------------------------------------------------------- /options/test/001_xrestormer_sr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/test/001_xrestormer_sr.yml -------------------------------------------------------------------------------- /options/test/002_xrestormer_denoise.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/test/002_xrestormer_denoise.yml -------------------------------------------------------------------------------- /options/test/003_xrestormer_deblur.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/test/003_xrestormer_deblur.yml -------------------------------------------------------------------------------- /options/test/004_xrestormer_derain.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/test/004_xrestormer_derain.yml -------------------------------------------------------------------------------- /options/test/005_xrestormer_dehaze.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/test/005_xrestormer_dehaze.yml -------------------------------------------------------------------------------- /options/train/001_xrestormer_s_sr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/001_xrestormer_s_sr.yml -------------------------------------------------------------------------------- /options/train/001_xrestormer_sr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/001_xrestormer_sr.yml -------------------------------------------------------------------------------- /options/train/002_xrestormer_denoise.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/002_xrestormer_denoise.yml -------------------------------------------------------------------------------- /options/train/003_xrestormer_deblur.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/003_xrestormer_deblur.yml -------------------------------------------------------------------------------- /options/train/004_xrestormer_derain.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/004_xrestormer_derain.yml -------------------------------------------------------------------------------- /options/train/005_xrestormer_dehaze.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/005_xrestormer_dehaze.yml -------------------------------------------------------------------------------- /options/train/A11_MPRNet_SR_bs16p192.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/A11_MPRNet_SR_bs16p192.yml -------------------------------------------------------------------------------- /options/train/A12_SwinIR_SR_bs16p192.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/A12_SwinIR_SR_bs16p192.yml -------------------------------------------------------------------------------- /options/train/A13_Uformer_SR_bs16p192.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/A13_Uformer_SR_bs16p192.yml -------------------------------------------------------------------------------- /options/train/A14_Restomer_SR_bs16p192.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/A14_Restomer_SR_bs16p192.yml -------------------------------------------------------------------------------- /options/train/A15_NAFNet_SR_bs16p192.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/options/train/A15_NAFNet_SR_bs16p192.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/prepare_example_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/scripts/prepare_example_data.py -------------------------------------------------------------------------------- /scripts/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/scripts/test.sh -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/setup.py -------------------------------------------------------------------------------- /xrestormer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/__init__.py -------------------------------------------------------------------------------- /xrestormer/archs/MPRNet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/archs/MPRNet_arch.py -------------------------------------------------------------------------------- /xrestormer/archs/NAFNet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/archs/NAFNet_arch.py -------------------------------------------------------------------------------- /xrestormer/archs/NAFSSR_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/archs/NAFSSR_arch.py -------------------------------------------------------------------------------- /xrestormer/archs/Uformer_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/archs/Uformer_arch.py -------------------------------------------------------------------------------- /xrestormer/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/archs/__init__.py -------------------------------------------------------------------------------- /xrestormer/archs/restormer_base_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/archs/restormer_base_arch.py -------------------------------------------------------------------------------- /xrestormer/archs/xrestormer_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/archs/xrestormer_arch.py -------------------------------------------------------------------------------- /xrestormer/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/data/__init__.py -------------------------------------------------------------------------------- /xrestormer/data/dehaze_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/data/dehaze_dataset.py -------------------------------------------------------------------------------- /xrestormer/data/gaussian_noise_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/data/gaussian_noise_dataset.py -------------------------------------------------------------------------------- /xrestormer/data/size_paired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/data/size_paired_dataset.py -------------------------------------------------------------------------------- /xrestormer/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/losses/__init__.py -------------------------------------------------------------------------------- /xrestormer/losses/example_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/losses/example_loss.py -------------------------------------------------------------------------------- /xrestormer/models/MPRNet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/models/MPRNet_model.py -------------------------------------------------------------------------------- /xrestormer/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/models/__init__.py -------------------------------------------------------------------------------- /xrestormer/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/models/lr_scheduler.py -------------------------------------------------------------------------------- /xrestormer/models/restormer_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/models/restormer_model.py -------------------------------------------------------------------------------- /xrestormer/models/uformer_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/models/uformer_model.py -------------------------------------------------------------------------------- /xrestormer/models/xrestormer_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/models/xrestormer_model.py -------------------------------------------------------------------------------- /xrestormer/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/test.py -------------------------------------------------------------------------------- /xrestormer/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/train.py -------------------------------------------------------------------------------- /xrestormer/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Andrew0613/X-Restormer/HEAD/xrestormer/version.py --------------------------------------------------------------------------------