sczhou/LEDNet

LEDNet: Joint Low-light Enhancement and Deblurring in the Dark (ECCV 2022)

Paper | Project Page | Video | Replicate Demo | TensorRT & ONNX

Shangchen Zhou, Chongyi Li, Chen Change Loy

S-Lab, Nanyang Technological University

Updates

• 2025.10.29: LEDNet_TensorRT provides both ONNXRuntime and TensorRT implementations. 🚀 Many thanks to @koamd!
• 2022.09.30: Upload LOLBlur dataset to BaiduPan (key: dz6u) for quick download in China :cn:
• 2022.09.27: Upload LOLBlur dataset to Google Drive.
• 2022.09.27: The code and pre-trained models are released :hugs:
• 2022.07.04: This repo is created.

TODO

• Built and train a new network for low-light enhancement
• Add checkpoint of LEDNetGAN (trained with GAN)
• Add checkpoint of retrained LEDNet (retrained on BasicSR)
• Add checkpoint of LEDNet in paper

LOL-Blur Dataset

(The datasets are hosted on both Google Drive and BaiduPan)

Dependencies and Installation

• Pytorch >= 1.7.1
• CUDA >= 10.1
• Other required packages in requirements.txt

# git clone this repository
git clone https://github.com/sczhou/LEDNet
cd LEDNet

# create new anaconda env
conda create -n lednet python=3.8 -y
conda activate lednet

# install python dependencies
pip3 install -r requirements.txt
python basicsr/setup.py develop

Train the Model

Before training, you need to:

• Download the LOL-Blur Dataset from Google Drive / BaiduPan (key: dz6u).
• Specify dataroot_gt and dataroot_lq in the corresponding option file.

Training LEDNet:

# without GAN
python basicsr/train.py -opt options/train_LEDNet.yml

# with GAN
python basicsr/train.py -opt options/train_LEDNetGAN.yml

This project is built on BasicSR, the detailed tutorial on training commands and config settings can be found here.

Quick Inference

• Download the LEDNet pretrained model from [Release V0.1.0] to the weights folder. You can manually download the pretrained models OR download by runing the following command.

  python scripts/download_pretrained_models.py LEDNet

Inference LEDNet:

# test LEDNet (paper model)
python inference_lednet.py --model lednet --test_path ./inputs

# test retrained LEDNet (higher PSNR and SSIM)
python inference_lednet.py --model lednet_retrain --test_path ./inputs

# test LEDNetGAN
python inference_lednet.py --model lednetgan --test_path ./inputs

The results will be saved in the results folder.

Evaluation

# set evaluation metrics of 'psnr', 'ssim', and 'lpips (vgg)'
python scripts/calculate_iqa_pair.py --result_path 'RESULT_ROOT' --gt_path 'GT_ROOT' --metrics psnr ssim lpips

(The released model was retrained using the BasicSR framework, which makes it easier to use or further develop upon this work. NOTE that the PSNR and SSIM scores of retrained model are higher than the paper model.)

Generate Low-light Images from Your Own Data

• Download the CE-ZeroDCE pretrained model from [Release V0.1.0] to the weights folder. You can manually download the pretrained models OR download by runing the following command.

  python scripts/download_pretrained_models.py CE-ZeroDCE

Run low-light generation:

python scripts/generate_low_light_imgs.py --test_path 'IMG_ROOT' --result_path 'RESULT_ROOT' --model_path './weights/ce_zerodce.pth'

Inference with Cog

To run containerized local inference with LEDNet using Cog, run the following commands in the project root:

cog run python basicsr/setup.py develop
cog predict -i image=@'path/to/input_image.jpg'

You can view this demo running as an API here on Replicate.

License

This project is licensed under S-Lab License 1.0. Redistribution and use for non-commercial purposes should follow this license.

Acknowledgement

This project is based on BasicSR. We calculate evaluation metrics using IQA-PyTorch toolbox. Thanks for their awesome works.

Citation

If our work is useful for your research, please consider citing:

@InProceedings{zhou2022lednet,
    author = {Zhou, Shangchen and Li, Chongyi and Loy, Chen Change},
    title = {LEDNet: Joint Low-light Enhancement and Deblurring in the Dark},
    booktitle = {ECCV},
    year = {2022}
}

Contact

If you have any questions, please feel free to reach me out at [email protected].