Unsupervised Low-Light Image Enhancement via Histogram Equalization Prior

• Python3
• PyTorch>=1.0
• NVIDIA GPU+CUDA

The original LOL dataset can be downloaded from here. The EnlightenGAN dataset can be downloaded from here Before starting training process, you should modify the data_root in ./config, and then run the following command

python LUM_train.py
python NDM_train.py

Please put test images into 'test_images' folder and download the pre-trained checkpoints from google drive(put it into ./checkpoints), then just run

python NDM_test.py

You can also just evaluate the stage one (LUM), just run

python LUM_test.py

HEP consists of two stages, Light Up Module (LUM) and Noise Disentanglement Module (LUM)

More visual results can be found in asssets.

if you find this repo is helpful, please cite

@article{zhang2021unsupervised,
  title={Unsupervised Low-Light Image Enhancement via Histogram Equalization Prior},
  author={Zhang, Feng and Shao, Yuanjie and Sun, Yishi and Zhu, Kai and Gao, Changxin and Sang, Nong},
  journal={arXiv preprint arXiv:2112.01766},
  year={2021}
}