[2020/11/05] Our code released.
Pytorch >= 1.1.0
numpy >= 1.6.0
tensorboardX
mkdir checkpoint data NH_results training_log- Download datasets into ./data folder
- Use
bash run.sh - If you train on REDIES datasets, please use train_phrase 1.
- If you train on NTIRE2020， please train train_phrase 1 first. Then, copy trained trained parameter to G1 and G2(crop_size is [240, 240]). Finetune G2. Train G3 finally(crop_size is [960, 960]). You can choose whether to do data augmentation by aug_fog.py.
If your find our research is helpful for you, please cite our paper.