Image Segmentation for Night-Vision Surveillance Camera Based on Deep Learning

Yuxiang Cao,Junhua Wang,Lizhe Qi,Min Xu
DOI: https://doi.org/10.1117/12.2654811
2022-01-01
Abstract:Under low illumination environments, the insufficient visible light and the existence of near-infrared light will cause photon noise and color distortion for the imaging of night-vision CMOS sensor. The light source highly affects the imaging of surveillance camera and declines the accuracy of semantic segmentation. In this work, we report a modified convolutional neural network based on DeepLabV3+. We modify the backbone of the network from Xception to MobileNetV2 to deal with the real-time vision task of night-vision surveillance camera. Linear bottleneck and inverted residuals are adopted in MobileNetV2, and they greatly reduce parameters of the network. A real-world low-light dataset with fine annotations for night-vision surveillance camera is proposed to train and evaluated the new framework. Aiming at the problem of insufficient training samples, transfer learning and a new image enhancement strategy are carried out to complete the training. We also change the loss function to a joint loss function to further improve the results of segmentation. Comparing with other existing state-of-the-art algorithms, the modified neural network shows competitive performance on both subjective and objective assessments. The ablation study comparing with the baseline model proves the effectiveness of the modifications.
What problem does this paper attempt to address?