yexuehua/DeepUPE

Underexposed Photo Enhancement Using Deep Illumination Estimation

Underexposed Photo Enhancement Using Deep Illumination Estimation

Ruixing Wang¹, Qing Zhang², Chi-Wing Fu¹, Xiaoyong Shen³, Wei-Shi Zheng², Jiaya Jia^1,3

¹The chinese university of hong kong ²Sun Yat-sen University ³Tencent Youtu Lab

Paper, Errata

Usage

1. Clone the repository:

   git clone https://github.com/wangruixing/DeepUPE.git

2. Install the Python dependencies, run:

   cd main
   pip install -r requirements.txt
   make

3. Evaluation: The test set can be downloaded in https://drive.google.com/file/d/1HZnNgptNxjKJAhekz2K5yh0mW0yKIws2/view?usp=sharing. It includes 500 pair images from MIT-Adobe FiveK 4500-5000. You can download this and run:

    python main/run.py checkpoints <input file path> <output file path>

PSNR evaluation code is in avg_psnr.m. Modify the related paths in 'avg_psnr.m', and run it.

Errata

We recently found an implementation bug in calculating PSNR. Fortunately, this bug doesn't affect any of the conclusions in our paper, we have corrected this bug in the Matlab code and updated the corresponding values in the revised paper. We apologize for the confusion to readers.

Bibtex

@InProceedings{Wang_2019_CVPR,
author = {Wang, Ruixing and Zhang, Qing and Fu, Chi-Wing and Shen, Xiaoyong and Zheng, Wei-Shi and Jia, Jiaya},
title = {Underexposed Photo Enhancement Using Deep Illumination Estimation},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2019}
}