A2-RL: Aesthetics Aware Reinforcement Learning for Image Cropping

  • 2018-03-12 15:49:14
  • Debang Li, Huikai Wu, Junge Zhang, Kaiqi Huang
  • 0


Image cropping aims at improving the aesthetic quality of images by adjustingtheir composition. Most weakly supervised cropping methods (without boundingbox supervision) rely on the sliding window mechanism. The sliding windowmechanism requires fixed aspect ratios and limits the cropping region witharbitrary size. Moreover, the sliding window method usually produces tens ofthousands of windows on the input image which is very time-consuming. Motivatedby these challenges, we firstly formulate the aesthetic image cropping as asequential decision-making process and propose a weakly supervised AestheticsAware Reinforcement Learning (A2-RL) framework to address this problem.Particularly, the proposed method develops an aesthetics aware reward functionwhich especially benefits image cropping. Similar to human's decision making,we use a comprehensive state representation including both the currentobservation and the historical experience. We train the agent using theactor-critic architecture in an end-to-end manner. The agent is evaluated onseveral popular unseen cropping datasets. Experiment results show that ourmethod achieves the state-of-the-art performance with much fewer candidatewindows and much less time compared with previous weakly supervised methods.


Introduction (beta)



Conclusion (beta)