Research on Multi-model Fusion Algorithm for Image Dehazing Based on Attention Mechanism

Abstract

AbstractIn recent years, the researchers of image dehazing mainly focused on deep learning algorithms. However, due to the defective network structure, and inadequate feature extraction, the deep learning algorithm still has many problems to be solved. In this paper, we fuse the physical models including haze imaging model with absorption compensation, multiple scattering imaging model and multi-scale retinex imaging model with convolutional neural network to construct the image dehazing network. Multiple scattering haze imaging model is used to describe the haze imaging process in a more consistent way with the physical imaging mechanism. And the multi-scale retinex imaging model ensures the color fidelity. In the network structure, multi-scale feature extraction module can improve network performance in terms of feature reuse. In the attention feature extraction module, the back-propagating of the important front features is used to enhance features. This method can effectively make up for the deficiency that autocorrelation features cannot share the deep-level information, which is also effective for features replenishment. The results of the comparative experiment demonstrate that our network outperforms state-of-the-art dehazing methods.KeywordsImage dehazingMulti-model fusionAbsorption compensationAttention mechanism