SRM-Net: An Effective End-to-end Neural Network for Single Image Dehazing

Yu Li,Wenbin Yu,Lingya Liu,Yiyin Wang,Lu Ding

DOI: https://doi.org/10.1145/3376067.3376081

2019-01-01

Abstract:Recently, the great development of deep learning has prompted many neural networks for single image dehazing to occur. However, due to the ill-posed nature of haze, an excellent characteristics representation capacity is still challenging. In this paper, we propose a lightweight yet effective senet-residual (SE-Res) multiscale end-to-end neural network named SRM-Net. Inspired by the remarkable performance of residual networks, we introduce a SE-Res structure which is an improved residual framework with an embedded SE unit to obtain feature maps. These maps pass through a multiscale mapping layer which can aggregate characteristics in different receptive fields. Notably, the utilization of all point-wise convolutions in the SRM-Net leads to fewer parameters for training, and the reuse of feature maps makes it more lightweight. Through extensive numerical experiments on three datasets including real hazy images, synthetic indoor and outdoor hazy images, the proposed SRM-Net achieves superior performances on subjective visual results and objective evaluation metrics compared to the state-of-the-art methods.