Skip to content

Official pytorch implementation for "Low-light Image Enhancement with Wavelet-based Diffusion Models"

License

Notifications You must be signed in to change notification settings

JianghaiSCU/Diffusion-Low-Light

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

[Siggraph Asia 2023]Low-light Image Enhancement with Wavelet-based Diffusion Models [Paper].

Hai Jiang1,2, Ao Luo2, Haoqiang Fan2, Songchen Han1, Shuaicheng Liu3,2

1.Sichuan University, 2.Megvii Technology,

3.University of Electronic Science and Technology of China

Pipeline

Dependencies

pip install -r requirements.txt

Download the raw training and evaluation datasets

Paired datasets

LOLv1 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]

LOLv2 dataset: Wenhan Yang, Haofeng Huang, Wenjing Wang, Shiqi Wang, and Jiaying Liu. "Sparse Gradient Regularized Deep Retinex Network for Robust Low-Light Image Enhancement", TIP, 2021. [Baiduyun (extracted code: l9xm)] [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)]

Unpaired datasets

Please refer to [Project Page of RetinexNet.]

Pre-trained Models

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

How to train?

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

python train.py  

How to test?

python evaluate.py

Visual comparison

Citation

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

@article{jiang2023low,
  title={Low-light image enhancement with wavelet-based diffusion models},
  author={Jiang, Hai and Luo, Ao and Fan, Haoqiang and Han, Songchen and Liu, Shuaicheng},
  journal={ACM Transactions on Graphics (TOG)},
  volume={42},
  number={6},
  pages={1--14},
  year={2023}
}

Acknowledgement

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

About

Official pytorch implementation for "Low-light Image Enhancement with Wavelet-based Diffusion Models"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages