Night Enhancement 项目教程

Night Enhancement 项目教程

night-enhancement [ECCV2022] "Unsupervised Night Image Enhancement: When Layer Decomposition Meets Light-Effects Suppression", https://arxiv.org/abs/2207.10564 项目地址: https://gitcode.com/gh_mirrors/ni/night-enhancement

1. 项目的目录结构及介绍

night-enhancement/
├── data/
│   ├── input/
│   └── output/
├── models/
│   ├── __init__.py
│   └── model.py
├── utils/
│   ├── __init__.py
│   └── utils.py
├── config/
│   └── config.yaml
├── main.py
├── requirements.txt
└── README.md

• data/: 存放输入和输出数据的目录。
  • input/: 存放输入图像的目录。
  • output/: 存放处理后图像的目录。
• models/: 存放模型定义的目录。
  • init.py: 初始化文件。
  • model.py: 定义了图像增强模型的文件。
• utils/: 存放工具函数的目录。
  • init.py: 初始化文件。
  • utils.py: 包含一些辅助函数的文件。
• config/: 存放配置文件的目录。
  • config.yaml: 项目的配置文件。
• main.py: 项目的启动文件。
• requirements.txt: 项目依赖的Python包列表。
• README.md: 项目的说明文档。

2. 项目的启动文件介绍

main.py 是项目的启动文件，负责加载配置、初始化模型、处理输入图像并保存输出图像。以下是 main.py 的主要功能：

• 加载配置文件 config.yaml。
• 初始化图像增强模型。
• 读取输入图像。
• 使用模型处理图像。
• 保存处理后的图像到输出目录。

3. 项目的配置文件介绍

config.yaml 是项目的配置文件，用于配置项目的各种参数。以下是 config.yaml 的主要内容：

model:
  name: "enhancement_model"
  parameters:
    alpha: 1.5
    beta: 0.5

data:
  input_dir: "data/input"
  output_dir: "data/output"

processing:
  mode: "auto"
  save_format: "png"

• model: 模型的配置。
  • name: 模型的名称。
  • parameters: 模型的参数。
    • alpha: 图像增强的参数。
    • beta: 图像增强的参数。
• data: 数据路径的配置。
  • input_dir: 输入图像的目录。
  • output_dir: 输出图像的目录。
• processing: 处理选项的配置。
  • mode: 处理模式，如 "auto" 或 "manual"。
  • save_format: 输出图像的保存格式，如 "png" 或 "jpg"。

night-enhancement [ECCV2022] "Unsupervised Night Image Enhancement: When Layer Decomposition Meets Light-Effects Suppression", https://arxiv.org/abs/2207.10564 项目地址: https://gitcode.com/gh_mirrors/ni/night-enhancement