Low illumination image enhancement based on attention mechanism and global illumination estimation

Abstract

Aiming at the problems of insufficient or excessive local brightness enhancement, color distortion, and excessive noise in the existing low-light image enhancement algorithms, a low-light image enhancement method combining attention mechanism and global illumination estimation is proposed. First, the illumination distribution map of low illumination image is obtained through the illumination distribution estimation network coupled with the attention gate mechanism. Then, the weight of the light distribution map is learned in the feature attention module. Finally, the image details are fused by the detail reconstruction module to create improved image. According to the experimental results, this method may effectively improve image brightness, contrast, and color in subjective visual effects while also enhancing objective evaluation indicators like PSNR, SSIM, and MSE when compared to some conventional methods.