Skip to content

Latest commit

 

History

History
49 lines (33 loc) · 1.8 KB

README.md

File metadata and controls

49 lines (33 loc) · 1.8 KB
Raw

MoESR: Blind Super-Resolution using Kernel-Aware Mixture of Experts

This repository is the official implementation of "MoESR: Blind Super-Resolution using Kernel-Aware Mixture of Experts".

Requirements

To install requirements:

conda env create -f environment.yml
conda activate MoESR_env

Datasets and pretrained models

You can download all datasets (DIV2KRK, Flickr2KRK and Urban100RK) and pretrained models from the following link: https://drive.google.com/drive/folders/1v7Lthkp-nLwdXGGkqolBgl5H88oNDH6l?usp=sharing

Test on synthetic datasets

For example to evaluate on DIV2KRK dataset:

cd codes
python main.py --in_dir ../datasets/DIV2KRK/lr_x2 --out_dir ../results/DIV2KRK/x2 --gt_dir ../datasets/DIV2KRK/gt --scale 2
python main.py --in_dir ../datasets/DIV2KRK/lr_x4 --out_dir ../results/DIV2KRK/x4 --gt_dir ../datasets/DIV2KRK/gt --scale 4

Test on real images

To evaluate on real-world images:

cd codes
python main.py --in_dir 'path to the LR input images' --out_dir 'path to save results' --scale 2 --real
python main.py --in_dir 'path to the LR input images' --out_dir 'path to save results' --scale 4 --real

Results

Our model achieves the following performance values (PSNR / SSIM) on DIV2KRK, Flickr2KRK and Urban100RK datasets:

Model name Scale DIV2KRK Flickr2KRK Urban100RK
MoESR x2 32.69 / 0.9054 32.95 / 0.9056 27.29 / 0.8448
MoESR x4 28.48 / 0.7805 28.57 / 0.7795 23.62 / 0.6766

Acknowledgement

The code is built on DualSR and KernelGAN. We thank the authors for sharing the codes.