zhengxinChenee / TTSR

CVPR 2020 TTSR: Learning Texture Transformer Network for Image Super-Resolution

TTSR

Official PyTorch implementation of the paper Learning Texture Transformer Network for Image Super-Resolution accepted in CVPR 2020.

Requirements and dependencies

• python 3.7 (recommend to use Anaconda)
• python packages: pip install numpy opencv-python
• pytorch >= 1.1.0
• torchvision >= 0.4.0

Model

Pre-trained models can be downloaded from onedrive, baidu cloud(0u6i), google drive.

• TTSR-rec.pt: trained with only reconstruction loss
• TTSR.pt: trained with all losses

Quick test

1. Clone this github repo

git clone https://github.com/FuzhiYang/TTSR.git
cd TTSR

2. Download pre-trained models and modify "model_path" in test.sh
3. Run test

sh test.sh

4. The results are in "save_dir" (default: ./test/demo/output)

Dataset prepare

1. Download CUFED train set and CUFED test set
2. Make dataset structure be:

• CUFED
  • train
    • input
    • ref
  • test
    • CUFED5

Evaluation

1. Prepare CUFED dataset and modify "dataset_dir" in eval.sh
2. Download pre-trained models and modify "model_path" in eval.sh
3. Run evaluation

sh eval.sh

4. The results are in "save_dir" (default: ./eval/CUFED/TTSR)

Train

1. Prepare CUFED dataset and modify "dataset_dir" in train.sh
2. Run training

sh train.sh

3. The training results are in "save_dir" (default: ./train/CUFED/TTSR)

Citation

@InProceedings{yang2020learning,
author = {Yang, Fuzhi and Yang, Huan and Fu, Jianlong and Lu, Hongtao and Guo, Baining},
title = {Learning Texture Transformer Network for Image Super-Resolution},
booktitle = {CVPR},
year = {2020},
month = {June}
}

Contact

If you meet any problems, please describe them in issues or contact:

• [e-mail]