Unpaired Image Enhancement Featuring Reinforcement-Learning-Controlled Image Editing Software
About
This paper tackles unpaired image enhancement, a task of learning a mapping function which transforms input images into enhanced images in the absence of input-output image pairs. Our method is based on generative adversarial networks (GANs), but instead of simply generating images with a neural network, we enhance images utilizing image editing software such as Adobe Photoshop for the following three benefits: enhanced images have no artifacts, the same enhancement can be applied to larger images, and the enhancement is interpretable. To incorporate image editing software into a GAN, we propose a reinforcement learning framework where the generator works as the agent that selects the software's parameters and is rewarded when it fools the discriminator. Our framework can use high-quality non-differentiable filters present in image editing software, which enables image enhancement with high performance. We apply the proposed method to two unpaired image enhancement tasks: photo enhancement and face beautification. Our experimental results demonstrate that the proposed method achieves better performance, compared to the performances of the state-of-the-art methods based on unpaired learning.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Low-light Image Enhancement | LOL v1 | PSNR15.23 | 113 | |
| Low-light Image Enhancement | LOL syn v2 | PSNR15.97 | 87 | |
| Low-light Image Enhancement | LOL real v2 | PSNR14.05 | 83 | |
| Image Enhancement | Image Enhancement Speed (test) | Running Time (ms)1.00e+4 | 56 | |
| Low-light Image Enhancement | SMID | PSNR23.11 | 34 | |
| Low-light Image Enhancement | SID | PSNR16.44 | 34 | |
| Low-light Image Enhancement | SDSD | PSNR20.97 | 30 | |
| Photo Retouching | FiveK 480p resolution (test) | PSNR22.11 | 27 | |
| Low-light Image Enhancement | SDSD-out | PSNR21.21 | 18 | |
| Image Enhancement | MIT-Adobe-5K-DPE (test) | PSNR22.27 | 13 |