Skip to content

Latest commit

 

History

History
52 lines (38 loc) · 980 Bytes

README.md

File metadata and controls

52 lines (38 loc) · 980 Bytes
Raw

Restormer

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

Network Architecture

Requirements

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