SRFormer 项目使用教程

SRFormer 项目使用教程

SRFormer Official code for "SRFormer: Permuted Self-Attention for Single Image Super-Resolution" (ICCV 2023) 项目地址: https://gitcode.com/gh_mirrors/sr/SRFormer

1. 项目目录结构及介绍

SRFormer/
├── basicsr/
│   ├── __init__.py
│   ├── ...
├── figs/
│   ├── ...
├── options/
│   ├── train/
│   │   ├── train_SRFormer_SRx2_scratch.yml
│   │   ├── train_SRFormer_SRx3_scratch.yml
│   │   ├── ...
│   ├── test/
│   │   ├── test_SRFormer_DF2Ksrx2.yml
│   │   ├── test_SRFormer_DF2Ksrx3.yml
│   │   ├── ...
├── scripts/
│   ├── dist_train.sh
│   ├── ...
├── tests/
│   ├── ...
├── .gitignore
├── LICENSE.txt
├── README.md
├── VERSION
├── requirements.txt
├── setup.cfg
├── setup.py

目录结构介绍

• basicsr/: 包含项目的主要代码文件，包括模型定义、训练和测试脚本等。
• figs/: 存放项目相关的图片文件。
• options/: 包含项目的配置文件，分为训练配置 (train/) 和测试配置 (test/)。
• scripts/: 包含项目的启动脚本，如分布式训练脚本 dist_train.sh。
• tests/: 存放项目的测试代码。
• .gitignore: Git 忽略文件配置。
• LICENSE.txt: 项目许可证文件。
• README.md: 项目说明文档。
• VERSION: 项目版本信息。
• requirements.txt: 项目依赖库列表。
• setup.cfg: 项目安装配置文件。
• setup.py: 项目安装脚本。

2. 项目启动文件介绍

启动文件

• scripts/dist_train.sh: 用于启动分布式训练的脚本。

使用方法

# 启动 SRFormer 的分布式训练
./scripts/dist_train.sh 4 options/train/SRFormer/train_SRFormer_SRx2_scratch.yml

参数说明

• 4: 表示使用 4 个 GPU 进行训练。
• options/train/SRFormer/train_SRFormer_SRx2_scratch.yml: 训练配置文件路径。

3. 项目配置文件介绍

配置文件目录

• options/train/: 训练配置文件目录。
• options/test/: 测试配置文件目录。

配置文件示例

训练配置文件 (train_SRFormer_SRx2_scratch.yml)

# 训练配置文件示例
name: SRFormer_SRx2_scratch
model: SRFormer
scale: 2
...

测试配置文件 (test_SRFormer_DF2Ksrx2.yml)

# 测试配置文件示例
name: test_SRFormer_DF2Ksrx2
model: SRFormer
scale: 2
...

配置文件说明

• name: 配置文件名称。
• model: 使用的模型名称。
• scale: 图像放大倍数。
• ...: 其他配置项，如数据集路径、优化器参数等。

通过以上配置文件，可以灵活地调整训练和测试的参数，以适应不同的任务需求。

SRFormer Official code for "SRFormer: Permuted Self-Attention for Single Image Super-Resolution" (ICCV 2023) 项目地址: https://gitcode.com/gh_mirrors/sr/SRFormer