Flexible Window-based Self-attention Transformer in Thermal Image Super-Resolution [Paper Link]
Hongcheng Jiang, Zhiqiang Chen
@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} }
Install Pytorh
pip install -r requirements.txt
- The test command is like
python predict.py
- Refer to
dataloader.pyfor dataset path to train. - Refer to
option.yamlfor the configuration file of the model to train. - The training command is like
python train.py
If you have any question, please email [email protected]