A Lightweight Dual-Branch Image Dehazing Network based on Associative Learning
DOI:
https://doi.org/10.54097/fcis.v4i2.9761Keywords:
Associative Learning, Encoder-Decoder, Image Dehazing, LightweightAbstract
Haze degrades the clarity, contrast, and details of images, resulting in a decrease in image quality. Image dehazing provides a means to obtain clearer and more accurate image information. Traditional methods for haze removal typically rely on manually designed features and models, limiting their performance in complex scenes. In recent years, the rapid advancement of deep learning has offered new insights into addressing the image dehazing problem. This paper proposes a lightweight dual-branch image dehazing network based on associative learning (LDANet). The network consists of a lightweight dehazing sub-network (LDSN) and a lightweight image enhancement subnetwork (LESN). To reduce computational and parameter complexity, the Tied Block Convolution (TBC) is employed, allowing for parameter sharing among components. Lastly, through associative learning, their distinctive features are mapped. Extensive experiments on synthetic and real-world datasets demonstrate the superiority of our approach in qualitative comparisons and quantitative evaluations compared to other state-of-the-art methods. Our method holds significant practical value for real-world image dehazing scenarios.
Downloads
References
W. E. K. Middleton. Vision through the atmosphere. University of Toronto Press, 1952.
Turgay Celik. Spatial entropy-based global and local image contrast enhancement[J]. IEEE Transactions on Image Processing, 2014, 23(12): 5298-5308.
STARK J A. Adaptive image contrast enhancement using generalizations of histogram equalization[J]. IEEE Transactions on Image Processing,2000,9(5):889-896.
KIM J Y,KIM L S,HWANG S H. An advanced contrast enhancement using partially overlapped sub-block histogram equalization [J]. IEEE Transactions on Circuits and Systems for Video Technology,2001,11(4):475-484.
Russo F. An image enhancement technique combining sharpening and noise reduction. IEEE Transactions on Instrumentation and Measurement, 2002, 51(4): 824−828.
Dippel S, Stahl M, Wiemker R, Blaffert T. Multiscale contrast enhancement for radiographies: Laplacian pyramid versus fast wavelet transform. IEEE Transactions on Medical Imaging, 2002, 21(4): 343−353.
Land E H, Mccann J J. Lightness and retinex theory[J]. Journal of the Optical Society of America, 1971,61(1):1-11.
He K M, Sun J, Tang X O. Single image haze removal using dark channel prior[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2011,33(12): 2341-2353.
Tan R T. Visibility in bad weather from a single image[C]. IEEE Conference on Computer Vision and Pattern Recognition, Anchorage, USA, 2008: 2347-2354.
Cai B L, Xu X M, Jia K, et al. DehazeNet: an endto-End system for single image haze removal[J]. IEEE Transactions on Image Processing, 2016, 25(11):5187-5198.
LI B Y,PENG X L,WANG Z Y,et al. AOD-net:All-in-one dehazing network[C]. IEEE International Conference on Computer Vision,2017:4770-4778.
REN W Q,LIU S,ZHANG H,et al. Single image dehazing via multiscale convolutional neural networks[C]. European Conference on Computer Vision,2016:154-169.
Chen D, He M, Fan Q, et al. Gated context aggregation network for image dehazing and deraining[C]. Proceedings of the 2019 IEEE winter conference on applications of computer vision. 2019: 1375-1383.
Qin X, Wang Z, Bai Y, et al. FFA-Net: Feature fusion attention network for single image dehazing[C]. Proceedings of the AAAI conference on artificial intelligence. 2020, 34(07): 11908-11915.
Lai, Wei Sheng , et al. Deep Laplacian Pyramid Networks for Fast and Accurate Super-Resolution [C]. IEEE Conference on Computer Vision and Pattern Recognition. IEEE Computer Society, 2017:5835-5843.
Wang Z , Bovik A C, Sheikh H R ,et al.Image quality assessment: from error visibility to structural similarity [J]. IEEE Trans Image Process, 2004, 13(4).
Li B, Ren W, Fu D, et al. Benchmarking single-image dehazing and beyond[J]. IEEE Transactions on Image Processing, 2018, 28 (1): 492-505.
Ancuti C, Ancuti C O, Timofte R, et al. I-HAZE: a dehazing benchmark with real hazy and haze-free indoor images[C]. Proceedings of the Advanced Concepts for Intelligent Vision Systems. 2018: 620-631.
Ancuti C O, Ancuti C, Timofte R, et al. O-haze: a dehazing benchmark with real hazy and haze-free outdoor images[C]. Proceedings of the IEEE conference on computer vision and pattern recognition workshops. 2018: 754-762.
Ancuti C O, Ancuti C, Timofte R. NH-HAZE: An image dehazing benchmark with non-homogeneous hazy and haze-free images[C]. Proceedings of the IEEE/CVF conference on computer vision and pattern recognition workshops. 2020: 444-445.
Li L, Zhou Y, Wu J, et al. Color-enriched gradient similarity for retouched image quality evaluation[J]. IEICE Transactions on Information and Systems, 2016, 99(3): 773-776.
Wang Z, Bovik A C, Sheikh H R, et al. Image quality assessment: from error visibility to structural similarity[J]. IEEE transactions on image processing, 2004, 13(4): 600-612.
Chen D, He M, Fan Q, et al. Gated context aggregation network for image dehazing and deraining[C]// Proceedings of the 2019 IEEE winter conference on applications of computer vision. 2019: 1375-1383.
Wei-Sheng Lai, Jia-Bin Huang, Narendra Ahuja, and Ming-Hsuan Yang. Deep laplacian pyramid networks for fast and accurate super-resolution. In CVPR,pages 624–632, 2017.
Di You, Jian Zhang, Jingfen Xie, Bin Chen, and Siwei Ma.COAST: Controllable arbitrary-sampling network for compressive sensing. IEEE Transactions on Image Processing, 30: 6066–6080, 2021.
