Restormer

A PyTorch implementation of Restormer based on CVPR 2022 paper Restormer: Efficient Transformer for High-Resolution Image Restoration.

Requirements

• Anaconda

• PyTorch

pip install -r requirements.txt

Dataset

Format:

|-- data     
    |-- SR
        |-- train
            |-- input
                sr-1.jpg
                ...
            `-- output
                sr-1.jpg
                ...
        `-- test                                                        
    |-- DN
        same as SR

Train Model

python train.py --data_name SR --seed 1

Test Model

python inference.py --data_name SR --model_file result/SR.pth

Continue training with checkpoint

python train.py --data_name SR --checkpoint_file result/checkpoint.pth