Low Light Image Enhancement Network With Attention Mechanism and Retinex Model

Abstract

Due to the uncertainty of the environment, the captured image may not only degrade but also have uneven brightness distribution. The quality of these images can not meet the input requirements of existing computer vision tasks, so they may lead to performance degradation in the completion of computer vision tasks. Past methods are often only applicable to low-light images with uniform brightness distribution, and the performance of these methods is not ideal for low-light images with uneven brightness distribution. In order to solve the problems caused by these low-light images, a new low light image enhancement model based on attention mechanism and Retinex model is proposed. The proposed method first estimates the illumination mask of the input image, which guides the network to predict the illumination distribution. Then we use a module with attention mechanism to predict the illumination map, and the initial enhanced image is estimated based on Retinex model. We modify the color distortion and suppress noise with convolution layers to obtain final enhanced results. In experiments, the performance of our methods is demonstrated by compared with the state-of-the-art existing methods. Our approach has more positive performance in some scenarios, especially uneven lighting.