基于小波变换及注意力机制的T型图像去雾网络

T-shaped Image Dehazing Network Based on Wavelet Transform and Attention Mechanism

受大气中雾霾等悬浮颗粒的影响,室外拍摄的图像常伴有低对比度和低能见度问题,现存去雾方法未能充分利用图像的局部特征信息,存在去雾不彻底及图像细节丢失等问题.为此,提出一种基于小波变换及注意力机制的T型图像去雾网络.所提网络通过对图像进行多次离散小波分解及重构来获取有雾图像的边缘细节特征,并提出了一种兼顾图像全局特征及局部信息提取的特征注意力模块,加强了网络在图像视觉感知和细节纹理方面的学习.在进行特征提取的过程中,提出T型连接方式来获得多尺度的图像特征,扩展了网络的表示能力.对重构后的无雾图像进行色彩平衡,得到最终复原图像.在合成数据集和真实数据集中的大量实验结果表明,本文所提网络相较于现有其他网络模型具有更优越的性能.

As affected by suspended particles such as haze in the atmosphere,images taken outdoors often suffer from low contrast and low visibility.The existing dehazing methods fail to make full use of the local feature information of the image,and there are problems such as incomplete dehazing and loss of image details.For this reason,this paper proposes a T-shaped image dehazing network based on wavelet transform and attention mechanism.Specifically,the proposed network obtains the edge detail features of the hazy image by performing multiple discrete wavelet decomposition and reconstruction on the image and proposes a feature attention module that takes into account both the global feature and the local information extraction of the image,which strengthens the network’s learning in image visual perception and detail texture.Secondly,in the process of feature extraction,a T-shaped method is proposed to obtain multi-scale image features,which expands the network’s representation ability.Finally,color balance is performed on the reconstructed clear image to obtain the final restored image.A large number of experimental results in synthetic data sets and real data sets show that the network proposed has superior performance when compared with other existing network models.