pip install -r requirements.txt

LOL dataset: Chen Wei, Wenjing Wang, Wenhan Yang, and Jiaying Liu. "Deep Retinex Decomposition for Low-Light Enhancement". BMVC, 2018. [Baiduyun (extracted code: sdd0)] [Google Drive]

LSRW dataset: Jiang Hai, Zhu Xuan, Ren Yang, Yutong Hao, Fengzhu Zou, Fang Lin, and Songchen Han. "R2RNet: Low-light Image Enhancement via Real-low to Real-normal Network". Journal of Visual Communication and Image Representation, 2023. [Baiduyun (extracted code: wmrr)]

Please refer to [Project Page of RetinexNet].

You can download our pre-trained model from [Google Drive] and [Baidu Yun (extracted code:cjzk)]

You need to modify datasets/dataset.py slightly for your environment, and then

python train.py  

python evaluate.py

If you use this code or ideas from the paper for your research, please cite our paper:

@InProceedings{Jiang_2024_ECCV,
    author    = {Jiang, Hai and Luo, Ao and Liu, Xiaohong and Han, Songchen and Liu, Shuaicheng},
    title     = {LightenDiffusion: Unsupervised Low-Light Image Enhancement with Latent-Retinex Diffusion Models},
    booktitle = {European Conference on Computer Vision},
    year      = {2024},
    pages     = {}
}

Part of the code is adapted from previous works: WeatherDiff and MIMO-UNet. We thank all the authors for their contributions.