Select & enhance: Masked-based image enhancement through tree-search theory and deep reinforcement learning

Marco Cotogni,Claudio Cusano
DOI: https://doi.org/10.1016/j.patrec.2024.05.013
IF: 4.757
2024-05-30
Pattern Recognition Letters
Abstract:The enhancement of low-quality images is both a challenging task and an essential endeavor in many fields including computer vision, computational photography, and image processing. In this paper, we propose a novel and fully explainable method for image enhancement that combines spatial selection and histogram equalization. Our approach leverages tree-search theory and deep reinforcement learning to iteratively select areas to be processed. Extensive experimentation on two datasets demonstrates the quality of our method compared to other state-of-the-art models. We also conducted a multi-user experiment which shows that our method can emulate a variety of enhancement styles. These results highlight the effectiveness and versatility of the proposed method in producing high-quality images through an explainable enhancement process.
computer science, artificial intelligence
What problem does this paper attempt to address?