Attention-Capsule Network for Low-Light Image Recognition.

Shiqi Shen,Zetao Jiang,Xiaochun Lei,Xu Wu,Yuting He
DOI: https://doi.org/10.1109/IJCNN54540.2023.10191264
2023-01-01
Abstract:Deep learning models have made extraordinary progress in recent years, but image recognition in low-light conditions has not been studied much. Due to its application prospects, low-light image recognition has still received focuses. Low-light image recognition is still challenging because of the crucial feature information that is hard to mine and learned by a deep learning model. Compared with enhancing the image's brightness before recognition, recognizing low-light images by end-to-end mode straightly has a more practical sense. In this paper, we propose a novel end-to-end model named Attention-Capsule Network (ACNet) for low-light image recognition tasks. The proposed model is extend from capsule structure, and its key component is the Global-Local Attention (GLA) module. The GLA module is designed to effectively and comprehensively combine important information. It mines global perception information by global block and gets local detail information from the particular local union. This strategy can significantly improve the performance of the proposed model. In addition, this paper proposes a directional learning loss, which guides the model to extract key features of images by optimizing the error between reconstructed images and normal-light images. Our experimental results demonstrate the effectiveness and robustness of our proposed attention-capsule Network model for low-light image recognition.
What problem does this paper attempt to address?