Attention-based end-to-end image defogging network

Aiming at the problem that the traditional prior information-based defogging algorithm fails in some special scenarios, an end-to-end convolutional defogging network based on attention mechanism is proposed. The network consists of two modules: parameter estimation and image restoration. First, multi-scale convolution is used to extract image feature information. Residual network and skip connection methods are used to improve the utilisation rate of shallow network feature information. Secondly, the channel domain attention is used to add weight to the feature image input from the previous network and select useful feature information. Finally, the atmospheric visibility model is combined to achieve image visibility restoration. The experimental results show that the proposed algorithm can effectively improve the visibility of the image and the restoration effect is natural. The objective evaluation index of the benchmark datasets also shows the effectiveness of the algorithm.