FAW-SAT

Flexible Window-based Self-attention Transformer in Thermal Image Super-Resolution [Paper Link]

Hongcheng Jiang, Zhiqiang Chen

Citations

BibTeX

@InProceedings{Jiang_2024_CVPR, author = {Jiang, Hongcheng and Chen, Zhiqiang}, title = {Flexible Window-based Self-attention Transformer in Thermal Image Super-Resolution}, booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops}, month = {June}, year = {2024}, pages = {3076-3085} }

Environment

• PyTorch >= 2.0
• BasicSR == 1.3.4.9

Installation

Install Pytorh

pip install -r requirements.txt

How To Test

• The test command is like

python predict.py 

How To Train

• Refer to dataloader.py for dataset path to train.
• Refer to option.yaml for the configuration file of the model to train.
• The training command is like

python train.py 

Contact

If you have any question, please email [email protected]