Uvcgan-Dehaze: a dehazing method for unpaired images

Canlin Li,Xiangfei Zhang,Wenjiao Zhang,Haowen Su,Lihua Bi
DOI: https://doi.org/10.1007/s00500-024-09924-4
IF: 3.732
2024-07-25
Soft Computing
Abstract:To solve the problems of feature loss and color difference after image dehazing and poor dehazing effect in real hazy images, a method UVCGAN-Dehaze is proposed for unpaired image dehazing. In the proposed model, the generator learns the relationship between image features and low-frequency features at multiple scales by combining the U-Net with the transformer. We strengthen the connection between spatial and channel features by adding Convolutional Block Attention Module(CBAM) before the skip connection of each layer of U-Net of the generator and adaptively adjusting the weight allocation of the feature map, so that the image after dehazing can better retain the detailed information of the image. Simultaneously, in order to reduce the feature loss and color difference after image dehazing, we propose a loss function named Feature Laplacian Perception (FLP) loss, which jointly constrains the generator by fusing the perception loss and Laplacian loss, so that the generated image reduces the feature loss and color difference. The method not only generates dehazing images with high perceptual quality in terms of important regions and details, but also avoids content and color loss. Extensive experiments verify the effectiveness of the method on synthetic images and real hazy images, and the proposed method achieves better dehazing results when compared with the experimental results of previous typical methods.
computer science, artificial intelligence, interdisciplinary applications
What problem does this paper attempt to address?