/contrast_enhancement_based_on_RL

(19th ICCWAMTIP) AUTOMATIC IMAGE CONTRAST ENHANCEMENT BASED ON REINFROCEMENT LEARNING

Primary LanguagePython

contrast_enhancement_based_on_RL

AUTOMATIC IMAGE CONTRAST ENHANCEMENT BASED ON REINFROCEMENT LEARNING

Abstract: Image contrast enhancement is a subjective problem depending on personal preference and subject field property. In this paper, we proposed a simple and efficient reinforcement learning based image contrast enhancement method for personal preference. Our method consists of state, action, reward or punishment definition, and policy learning. We have implemented Q-learning and State Action Reward State Action(SARSA) algorithms. The training process is easy for any user by clicking some buttons in our developed graphical user interface. The experimental results demonstrate good performance of our proposed method in this paper.

https://ieeexplore.ieee.org/abstract/document/10016571

result

First algorithm on the training first_algorithm_training

Second algorithm on the training second_alg

training evaluation evaluation

Test result test_img