Distort-and-Recover: Color Enhancement using Deep Reinforcement Learning

  • 2018-04-16 01:48:00
  • Jongchan Park, Joon-Young Lee, Donggeun Yoo, In So Kweon
  • 0

Abstract

Learning-based color enhancement approaches typically learn to map from inputimages to retouched images. Most of existing methods require expensive pairs ofinput-retouched images or produce results in a non-interpretable way. In thispaper, we present a deep reinforcement learning (DRL) based method for colorenhancement to explicitly model the step-wise nature of human retouchingprocess. We cast a color enhancement process as a Markov Decision Process whereactions are defined as global color adjustment operations. Then we train ouragent to learn the optimal global enhancement sequence of the actions. Inaddition, we present a 'distort-and-recover' training scheme which onlyrequires high-quality reference images for training instead of input andretouched image pairs. Given high-quality reference images, we distort theimages' color distribution and form distorted-reference image pairs fortraining. Through extensive experiments, we show that our method producesdecent enhancement results and our DRL approach is more suitable for the'distort-and-recover' training scheme than previous supervised approaches.Supplementary material and code are available athttps://sites.google.com/view/distort-and-recover/

 

Quick Read (beta)

loading the full paper ...