PFT-SR
CVPR2025-Progressive Focused Transformer for Single Image Super-Resolution
File Explorer
Download Latest Version (.zip)- __init__.cpython-312.pyc
- __init__.cpython-39.pyc
- test.cpython-39.pyc
- train.cpython-39.pyc
- version.cpython-39.pyc
- __init__.cpython-312.pyc
- __init__.cpython-39.pyc
- arch_util.cpython-312.pyc
- arch_util.cpython-39.pyc
- pft_arch.cpython-312.pyc
- pft_arch.cpython-39.pyc
- __init__.py
- arch_util.py
- pft_arch.py
- __init__.cpython-39.pyc
- data_sampler.cpython-39.pyc
- data_util.cpython-39.pyc
- paired_image_dataset.cpython-39.pyc
- prefetch_dataloader.cpython-39.pyc
- transforms.cpython-39.pyc
- __init__.py
- data_sampler.py
- data_util.py
- degradations.py
- paired_image_dataset.py
- prefetch_dataloader.py
- transforms.py
- __init__.cpython-39.pyc
- basic_loss.cpython-39.pyc
- loss_util.cpython-39.pyc
- __init__.py
- basic_loss.py
- loss_util.py
- __init__.cpython-39.pyc
- metric_util.cpython-39.pyc
- psnr_ssim.cpython-39.pyc
- test_psnr_ssim.py
- __init__.py
- metric_util.py
- psnr_ssim.py
- __init__.cpython-39.pyc
- base_model.cpython-39.pyc
- lr_scheduler.cpython-39.pyc
- pft_model.cpython-39.pyc
- sr_model.cpython-39.pyc
- __init__.py
- base_model.py
- lr_scheduler.py
- pft_model.py
- sr_model.py
- __init__.cpython-312.pyc
- __init__.cpython-39.pyc
- color_util.cpython-312.pyc
- color_util.cpython-39.pyc
- diffjpeg.cpython-312.pyc
- diffjpeg.cpython-39.pyc
- dist_util.cpython-312.pyc
- dist_util.cpython-39.pyc
- file_client.cpython-312.pyc
- file_client.cpython-39.pyc
- img_process_util.cpython-312.pyc
- img_process_util.cpython-39.pyc
- img_util.cpython-312.pyc
- img_util.cpython-39.pyc
- logger.cpython-312.pyc
- logger.cpython-39.pyc
- misc.cpython-312.pyc
- misc.cpython-39.pyc
- options.cpython-312.pyc
- options.cpython-39.pyc
- registry.cpython-312.pyc
- registry.cpython-39.pyc
- __init__.py
- color_util.py
- create_lmdb.py
- diffjpeg.py
- dist_util.py
- file_client.py
- flow_util.py
- img_process_util.py
- img_util.py
- lmdb_util.py
- logger.py
- matlab_functions.py
- misc.py
- options.py
- plot_util.py
- registry.py
- __init__.py
- test.py
- train.py
- version.py
- README.md
- attention_distributions.png
- classical.png
- lightweight.png
- PFT.png
- pft_m.png
- visual_classical.png
- visual_lightweight.png
- smm_cuda.cu
- smm_cuda.h
- make.sh
- setup.py
- test_smm_AmV.py
- test_smm_QmK.py
- 001_PFT_SRx2_scratch.yml
- 002_PFT_SRx3_finetune.yml
- 003_PFT_SRx4_finetune.yml
- 101_PFT_light_SRx2_scratch.yml
- 102_PFT_light_SRx3_finetune.yml
- 103_PFT_light_SRx4_finetune.yml
- 001_PFT_SRx2_scratch.yml
- 002_PFT_SRx3_finetune.yml
- 003_PFT_SRx4_finetune.yml
- 101_PFT_light_SRx2_scratch.yml
- 102_PFT_light_SRx3_finetune.yml
- 103_PFT_light_SRx4_finetune.yml
- .gitignore
- inference.py
- inference_image.png
- LICENSE.txt
- README.md
- requirements.txt
- setup.py
- VERSION
- VisualAttention.py
# Installation Guide
1. Get the code
git clone https://github.com/CVL-UESTC/PFT-SR
Downloads the entire project code from GitHub to your computer.
cd PFT-SR
Moves into the project folder you just downloaded.
2. Python
Easy RecommendedPrerequisites
pip install -r requirements.txt
Installs the Python libraries listed in requirements.txt (or similar).
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --use-env --nproc_per_node=8 --master_port=1145 basicsr/train.py -opt options/train/001_PFT_SRx2_scratch.yml --launcher pytorch
Runs the Python script (or module).
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --use-env --nproc_per_node=8 --master_port=1145 basicsr/train.py -opt options/train/003_PFT_SRx4_finetune.yml --launcher pytorch
Runs the Python script (or module).
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --use-env --nproc_per_node=4 --master_port=1145 basicsr/train.py -opt options/train/101_PFT_light_SRx2_scratch.yml --launcher pytorch
Runs the Python script (or module).
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --use-env --nproc_per_node=4 --master_port=1145 basicsr/train.py -opt options/train/102_PFT_light_SRx3_finetune.yml --launcher pytorch
Runs the Python script (or module).
If it runs without errors and prints output in the terminal, it worked.
Pulled directly from this repo's README.
// repository documentation
Was this content helpful?
(0 ratings)