AGLLNet : A Machine Learning Model for Brightening Dark Images
This is an introduction to「AGLLNet」, a machine learning model that can be used with ailia SDK. You can easily use this model to create AI applications using ailia SDK as well as many other ready-to-use ailia MODELS.
Overview
AGLLNet is a machine learning model that uses attention to correct exposure and brighten dark images. It was developed by Beihang University and published in August 2019.
Architecture
Brightening a dark image is a challenging task because it requires not only correcting the brightness, but also correcting the color distortion and noise hidden in the dark areas of the image. Simply adjusting the brightness make these distortions become apparent.
The first step to train this model was to create a data set. The authors took images with normal exposure as input, corrected the exposure to make it darker, and added noise. The area where exposure compensation is applied is called the ue-attention map, and the area where noise is added is called the noise map. Each image was also applied contrast and edge correction and the result was used as reference images.
AGLLNet uses an end-to-end attention-guided method based on multi-branch CNN. Train two attention maps for brightness correction and denoising on the new dataset we created. Using the newly created dataset for training, the method learns two attention maps to guide the brightness enhancement and denoising tasks respectively.
The first attention map recognizes underexposed areas, and the second recognizes noise in the texture. Based on these attention maps, a multi-branch correction CNN is applied.
The CNN applied to input images is made of 4 subnets. Attention-Net and Noise-Net are applied to guide the attention to underexposed areas and the denoising process. Then comes Enhancement-Net and Refinforce-Net to perform image enhancements. Enhancement-Net includes a feature extraction module, a enhancement module, and a fusion module to perform enhancing and denoising simultaneously. Lastly, ReinforceNet is designed for contrast re-enhancement to solve the low-contrast limitation caused by regression.
This method quantitatively outperforms conventional image enhancement models.
By applying this low-light image enhancement model as a preprocessing step for object detection, we can increase the detection rate of MaskRCNN.
Usage
AGLLNet can be used with ailia SDK using the following command.
$ python3 agllnet.py --input input.jpg --savepath output.jpg
ax Inc. has developed ailia SDK, which enables cross-platform, GPU-based rapid inference.
ax Inc. provides a wide range of services from consulting and model creation, to the development of AI-based applications and SDKs. Feel free to contact us for any inquiry.