MBLLEN 项目安装和配置指南

MBLLEN 项目安装和配置指南

MBLLEN Code for “MBLLEN: Low-light Image/Video Enhancement Using CNNs”, BMVC 2018. 项目地址: https://gitcode.com/gh_mirrors/mb/MBLLEN

1. 项目基础介绍和主要编程语言

项目介绍

MBLLEN（Multi-Branch Low-light Enhancement Network）是一个用于低光图像和视频增强的开源项目。该项目基于卷积神经网络（CNN），旨在提高在低光条件下拍摄的图像和视频的质量。MBLLEN 在 BMVC 2018 会议上发表，由 Feifan Lv、Feng Lu、Jianhua Wu 和 Chongsoon Lim 开发。

主要编程语言

该项目主要使用 Python 编程语言。

2. 项目使用的关键技术和框架

关键技术

• 卷积神经网络（CNN）：用于图像和视频的增强处理。
• Keras：用于构建和训练神经网络模型。
• TensorFlow：作为 Keras 的后端，用于深度学习计算。
• OpenCV：用于图像处理和预处理。

框架

• Keras：版本 2.2.0
• TensorFlow：版本 1.6.0
• OpenCV-Python：版本 3.4.2

3. 项目安装和配置的准备工作和详细安装步骤

准备工作

在开始安装之前，请确保您的系统已经安装了以下软件和库：

• Python 3.x
• Git
• pip（Python 包管理工具）

详细安装步骤

步骤 1：克隆项目仓库

首先，使用 Git 克隆 MBLLEN 项目到您的本地机器。

git clone https://github.com/Lvfeifan/MBLLEN.git
cd MBLLEN

步骤 2：创建虚拟环境（可选）

为了隔离项目的依赖环境，建议创建一个 Python 虚拟环境。

python3 -m venv mblen_env
source mblen_env/bin/activate  # 在 Windows 上使用 `mblen_env\Scripts\activate`

步骤 3：安装依赖库

使用 pip 安装项目所需的依赖库。

pip install -r requirements.txt

步骤 4：下载预训练模型

项目提供了多个预训练模型，您可以根据需要下载并放置在 models 目录下。

mkdir models
# 下载模型文件并放置在 models 目录下

步骤 5：测试模型

您可以使用提供的测试脚本来测试模型。

cd main
python test.py -i <input_folder> -r <output_folder> -m <model_name>

• input_folder：包含低光图像的文件夹路径。
• output_folder：输出增强后图像的文件夹路径。
• model_name：使用的预训练模型名称。

步骤 6：训练模型（可选）

如果您有自定义数据集，可以进行模型训练。

cd main
python train.py

确保您已经准备好了训练数据，并修改了 train.py 和 data_load.py 中的路径设置。

结语

通过以上步骤，您应该能够成功安装和配置 MBLLEN 项目，并开始使用它来增强低光图像和视频。如果您在安装过程中遇到任何问题，请参考项目的 GitHub 页面或相关文档。

MBLLEN Code for “MBLLEN: Low-light Image/Video Enhancement Using CNNs”, BMVC 2018. 项目地址: https://gitcode.com/gh_mirrors/mb/MBLLEN