CycleISP 项目使用教程

CycleISP 项目使用教程

CycleISP [CVPR 2020--Oral] CycleISP: Real Image Restoration via Improved Data Synthesis 项目地址: https://gitcode.com/gh_mirrors/cy/CycleISP

1. 项目目录结构及介绍

CycleISP/
├── dataloaders/
│   ├── __init__.py
│   └── ...
├── datasets/
│   ├── __init__.py
│   └── ...
├── networks/
│   ├── __init__.py
│   └── ...
├── pretrained_models/
│   ├── isp/
│   └── denoising/
├── utils/
│   ├── __init__.py
│   └── ...
├── LICENSE.md
├── README.md
├── generate_raw_data.py
├── generate_rgb_data.py
├── test_dnd_raw.py
├── test_dnd_rgb.py
├── test_sidd_raw.py
└── test_sidd_rgb.py

目录结构说明

• dataloaders/: 包含数据加载相关的代码。
• datasets/: 包含数据集相关的代码。
• networks/: 包含神经网络模型的代码。
• pretrained_models/: 包含预训练模型的文件，分为 isp 和 denoising 两个子目录。
• utils/: 包含项目中使用的各种工具函数。
• LICENSE.md: 项目的许可证文件。
• README.md: 项目的说明文档。
• generate_raw_data.py: 用于生成 RAW 数据对的脚本。
• generate_rgb_data.py: 用于生成 sRGB 数据对的脚本。
• test_dnd_raw.py: 用于测试 DND 数据集 RAW 图像的脚本。
• test_dnd_rgb.py: 用于测试 DND 数据集 sRGB 图像的脚本。
• test_sidd_raw.py: 用于测试 SIDD 数据集 RAW 图像的脚本。
• test_sidd_rgb.py: 用于测试 SIDD 数据集 sRGB 图像的脚本。

2. 项目启动文件介绍

generate_raw_data.py

该脚本用于生成 RAW 数据对。它将输入的干净 sRGB 图像转换为干净的 RAW 图像，并通过噪声注入模块添加不同级别的噪声，生成对应的噪声图像对。

使用方法:

python generate_raw_data.py

generate_rgb_data.py

该脚本用于生成 sRGB 数据对。它将输入的合成 RAW 噪声图像转换为噪声 sRGB 图像，从而生成 sRGB 噪声图像对。

使用方法:

python generate_rgb_data.py

3. 项目的配置文件介绍

CycleISP 项目没有明确的配置文件，但可以通过命令行参数或环境变量来配置一些运行时参数。例如，在 generate_raw_data.py 和 generate_rgb_data.py 中，可以通过命令行参数指定输入文件路径、输出文件路径等。

示例:

python generate_raw_data.py --input_path /path/to/input --output_path /path/to/output

通过这种方式，用户可以根据需要自定义项目的运行配置。

CycleISP [CVPR 2020--Oral] CycleISP: Real Image Restoration via Improved Data Synthesis 项目地址: https://gitcode.com/gh_mirrors/cy/CycleISP