Retinexformer 项目使用教程

1. 项目目录结构及介绍

Retinexformer 项目的目录结构如下：

Retinexformer/
├── basicsr/
│   ├── __init__.py
│   ├── train.py
│   └── ...
├── Enhancement/
│   ├── __init__.py
│   ├── test_from_dataset.py
│   └── ...
├── Options/
│   ├── RetinexFormer_LOL_v1.yml
│   ├── RetinexFormer_LOL_v2_real.yml
│   └── ...
├── pretrained_weights/
│   ├── LOL_v1.pth
│   ├── LOL_v2_real.pth
│   └── ...
├── data/
│   ├── LOLv1/
│   ├── LOLv2/
│   └── ...
├── README.md
├── LICENSE.txt
├── setup.cfg
├── setup.py
└── ...

目录结构介绍

• basicsr/: 包含项目的基础代码，如训练脚本 train.py 等。
• Enhancement/: 包含用于图像增强的代码，如测试脚本 test_from_dataset.py 等。
• Options/: 包含项目的配置文件，如 RetinexFormer_LOL_v1.yml 等。
• pretrained_weights/: 包含预训练模型的权重文件，如 LOL_v1.pth 等。
• data/: 包含项目所需的数据集，如 LOLv1、LOLv2 等。
• README.md: 项目的介绍文档。
• LICENSE.txt: 项目的许可证文件。
• setup.cfg 和 setup.py: 项目的安装配置文件。

2. 项目启动文件介绍

2.1 训练脚本 (train.py)

train.py 是项目的训练脚本，用于训练 Retinexformer 模型。启动训练的命令如下：

python basicsr/train.py --opt Options/RetinexFormer_LOL_v1.yml

2.2 测试脚本 (test_from_dataset.py)

test_from_dataset.py 是项目的测试脚本，用于对数据集进行测试。启动测试的命令如下：

python Enhancement/test_from_dataset.py --opt Options/RetinexFormer_LOL_v1.yml --weights pretrained_weights/LOL_v1.pth --dataset LOL_v1

3. 项目配置文件介绍

3.1 配置文件结构

配置文件位于 Options/ 目录下，常见的配置文件包括：

• RetinexFormer_LOL_v1.yml
• RetinexFormer_LOL_v2_real.yml
• RetinexFormer_LOL_v2_synthetic.yml

3.2 配置文件示例

以 RetinexFormer_LOL_v1.yml 为例，配置文件内容如下：

# 数据集路径
dataset_path: "data/LOLv1"

# 模型参数
model:
  name: "RetinexFormer"
  params:
    num_layers: 12
    num_heads: 8

# 训练参数
train:
  batch_size: 8
  num_epochs: 50
  learning_rate: 0.0002

# 测试参数
test:
  batch_size: 1

3.3 配置文件说明

• dataset_path: 指定数据集的路径。
• model: 定义模型的名称和参数，如 num_layers 和 num_heads。
• train: 定义训练时的参数，如 batch_size、num_epochs 和 learning_rate。
• test: 定义测试时的参数，如 batch_size。

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