Goal Conditioned Reinforcement Learning for Photo Finishing Tuning

  • 2025-03-10 14:20:59
  • Jiarui Wu, Yujin Wang, Lingen Li, Zhang Fan, Tianfan Xue
  • 0

Abstract

Photo finishing tuning aims to automate the manual tuning process of thephoto finishing pipeline, like Adobe Lightroom or Darktable. Previous workseither use zeroth-order optimization, which is slow when the set of parametersincreases, or rely on a differentiable proxy of the target finishing pipeline,which is hard to train. To overcome these challenges, we propose a novelgoal-conditioned reinforcement learning framework for efficiently tuningparameters using a goal image as a condition. Unlike previous approaches, ourtuning framework does not rely on any proxy and treats the photo finishingpipeline as a black box. Utilizing a trained reinforcement learning policy, itcan efficiently find the desired set of parameters within just 10 queries,while optimization based approaches normally take 200 queries. Furthermore, ourarchitecture utilizes a goal image to guide the iterative tuning of pipelineparameters, allowing for flexible conditioning on pixel-aligned target images,style images, or any other visually representable goals. We conduct detailedexperiments on photo finishing tuning and photo stylization tuning tasks,demonstrating the advantages of our method. Project website:https://openimaginglab.github.io/RLPixTuner/.

 

Quick Read (beta)

loading the full paper ...