LLIE

A repository for Low-Light Image Enhancement codes.

Usage

Install Requirements

1. install pixi (on linux)

curl -fsSL https://pixi.sh/install.sh | sh

Please refer to https://pixi.sh/latest/installation/ for more information.

2. clone the repo

git clone https://github.com/IgnorantCopy/LLIE.git
cd LLIE

3. install dependencies

pixi install --environment llie-gpu

If you do not have a gpu on your machine, please use llie-cpu instead of llie-gpu.

Download Datasets

The original links of the datasets are provided: LOLv1, LOLv2, DICM, LIME, MEF, NPE, VV, LOLBlur, SICE (code: yixu), SID (code: yixu), EnlightenGAN dataset, ZeroDCE dataset,

You can put the downloaded datasets in the same folder, and run the script to replace all the data root in config files:

export PYTHONPATH=$(pwd)
cd src/llie
pixi run --environment llie-gpu python replace_data_root.py --root /path/to/dataroot

Train a Model

pixi run --environment llie-gpu python train.py --config ./configs/method/config.yaml

Test a Model

pixi run --environment llie-gpu python test.py  --ckpt path/to/checkpoint.ckpt --config ./configs/test/config.yaml