zhaohengyuan1 / Pan
Programming Languages
Projects that are alternatives of or similar to Pan
If you have questions about results, please check the new update version of file PAN_arch.py.
PAN [⚡️ 272K parameters]
Lowest parameters in AIM2020 Efficient Super Resolution.
Paper | Video
Efficient Image Super-Resolution Using Pixel Attention
Authors: Hengyuan Zhao, Xiangtao Kong, Jingwen He, Yu Qiao, Chao Dong
Dependencies
- Python >= 3.6 (Recommend to use Anaconda)
- PyTorch >= 1.5.0
- NVIDIA GPU + CUDA
- Python packages:
pip install numpy opencv-python lmdb
- [option] Python packages:
pip install tensorboardX
, for visualizing curves.
Codes
- Our codes version based on mmsr.
- This codes provide the testing and training code.
How to Test
- Clone this github repo.
git clone https://github.com/zhaohengyuan1/PAN.git
cd PAN
-
Download the five test datasets (Set5, Set14, B100, Urban100, Manga109) from Google Drive
-
Pretrained models have be placed in
./experiments/pretrained_models/
folder. More models can be download from Google Drive. -
Run test. We provide
x2,x3,x4
pretrained models.
cd codes
python test.py -opt option/test/test_PANx4.yml
More testing commond can be found in ./codes/run_scripts.sh
file.
5. The output results will be sorted in ./results
. (We have been put our testing log file in ./results
) We also provide our testing results on five benchmark datasets on Google Drive or Baidu Drive, password: 8mrn.
How to Train
-
Download DIV2K and Flickr2K from Google Drive or Baidu Drive
-
Generate Training patches. Modified the path of your training datasets in
./codes/data_scripts/extract_subimages.py
file. -
Run Training.
python train.py -opt options/train/train_PANx4.yml
- More training commond can be found in
./codes/run_scripts.sh
file.
Testing the Parameters, Mult-Adds and Running Time
- Testing the parameters and Mult-Adds.
python test_summary.py
- Testing the Running Time.
python test_running_time.py
Related Work on AIM2020
Enhanced Quadratic Video Interpolation (winning solution of AIM2020 VTSR Challenge) paper | code
Contact
Email: [email protected]
If you find our work is useful, please kindly cite it.
@misc{zhao2020efficient,
title={Efficient Image Super-Resolution Using Pixel Attention},
author={Hengyuan Zhao and Xiangtao Kong and Jingwen He and Yu Qiao and Chao Dong},
year={2020},
eprint={2010.01073},
archivePrefix={arXiv},
primaryClass={eess.IV}
}