An Efficient Dehazing Algorithm Based on the Fusion of Transformer and Convolutional Neural Network

The purpose of image dehazing is to remove the interference from weather factors in degraded images and enhance the clarity and color saturation of images to maximize the restoration of useful features. Single image dehazing is one of the most important tasks in the field of image restoration. In re...

Full description

Bibliographic Details
Main Authors: Jun Xu, Zi-Xuan Chen, Hao Luo, Zhe-Ming Lu
Format: Article
Language:English
Published: MDPI AG 2022-12-01
Series:Sensors
Subjects:
Online Access:https://www.mdpi.com/1424-8220/23/1/43
_version_ 1797431205497405440
author Jun Xu
Zi-Xuan Chen
Hao Luo
Zhe-Ming Lu
author_facet Jun Xu
Zi-Xuan Chen
Hao Luo
Zhe-Ming Lu
author_sort Jun Xu
collection DOAJ
description The purpose of image dehazing is to remove the interference from weather factors in degraded images and enhance the clarity and color saturation of images to maximize the restoration of useful features. Single image dehazing is one of the most important tasks in the field of image restoration. In recent years, due to the progress of deep learning, single image dehazing has made great progress. With the success of Transformer in advanced computer vision tasks, some research studies also began to apply Transformer to image dehazing tasks and obtained surprising results. However, both the deconvolution-neural-network-based dehazing algorithm and Transformer based dehazing algorithm magnify their advantages and disadvantages separately. Therefore, this paper proposes a novel Transformer–Convolution fusion dehazing network (TCFDN), which uses Transformer’s global modeling ability and convolutional neural network’s local modeling ability to improve the dehazing ability. In the Transformer–Convolution fusion dehazing network, the classic self-encoder structure is used. This paper proposes a Transformer–Convolution hybrid layer, which uses an adaptive fusion strategy to make full use of the Swin-Transformer and convolutional neural network to extract and reconstruct image features. On the basis of previous research, this layer further improves the ability of the network to remove haze. A series of contrast experiments and ablation experiments not only proved that the Transformer–Convolution fusion dehazing network proposed in this paper exceeded the more advanced dehazing algorithm, but also provided solid and powerful evidence for the basic theory on which it depends.
first_indexed 2024-03-09T09:41:32Z
format Article
id doaj.art-3e722f89027d45909ac73cb1c128e363
institution Directory Open Access Journal
issn 1424-8220
language English
last_indexed 2024-03-09T09:41:32Z
publishDate 2022-12-01
publisher MDPI AG
record_format Article
series Sensors
spelling doaj.art-3e722f89027d45909ac73cb1c128e3632023-12-02T00:52:50ZengMDPI AGSensors1424-82202022-12-012314310.3390/s23010043An Efficient Dehazing Algorithm Based on the Fusion of Transformer and Convolutional Neural NetworkJun Xu0Zi-Xuan Chen1Hao Luo2Zhe-Ming Lu3Wenzhou Mass Transit Railway Investment Group Co., Ltd., Wenzhou 325000, ChinaSchool of Aeronautics and Astronautics, Zhejiang University, Hangzhou 310027, ChinaSchool of Aeronautics and Astronautics, Zhejiang University, Hangzhou 310027, ChinaSchool of Aeronautics and Astronautics, Zhejiang University, Hangzhou 310027, ChinaThe purpose of image dehazing is to remove the interference from weather factors in degraded images and enhance the clarity and color saturation of images to maximize the restoration of useful features. Single image dehazing is one of the most important tasks in the field of image restoration. In recent years, due to the progress of deep learning, single image dehazing has made great progress. With the success of Transformer in advanced computer vision tasks, some research studies also began to apply Transformer to image dehazing tasks and obtained surprising results. However, both the deconvolution-neural-network-based dehazing algorithm and Transformer based dehazing algorithm magnify their advantages and disadvantages separately. Therefore, this paper proposes a novel Transformer–Convolution fusion dehazing network (TCFDN), which uses Transformer’s global modeling ability and convolutional neural network’s local modeling ability to improve the dehazing ability. In the Transformer–Convolution fusion dehazing network, the classic self-encoder structure is used. This paper proposes a Transformer–Convolution hybrid layer, which uses an adaptive fusion strategy to make full use of the Swin-Transformer and convolutional neural network to extract and reconstruct image features. On the basis of previous research, this layer further improves the ability of the network to remove haze. A series of contrast experiments and ablation experiments not only proved that the Transformer–Convolution fusion dehazing network proposed in this paper exceeded the more advanced dehazing algorithm, but also provided solid and powerful evidence for the basic theory on which it depends.https://www.mdpi.com/1424-8220/23/1/43image dehazingdeep learningconvolutional neural networkTransformerTransformer–Convolution fusion dehazing network (TCFDN)
spellingShingle Jun Xu
Zi-Xuan Chen
Hao Luo
Zhe-Ming Lu
An Efficient Dehazing Algorithm Based on the Fusion of Transformer and Convolutional Neural Network
Sensors
image dehazing
deep learning
convolutional neural network
Transformer
Transformer–Convolution fusion dehazing network (TCFDN)
title An Efficient Dehazing Algorithm Based on the Fusion of Transformer and Convolutional Neural Network
title_full An Efficient Dehazing Algorithm Based on the Fusion of Transformer and Convolutional Neural Network
title_fullStr An Efficient Dehazing Algorithm Based on the Fusion of Transformer and Convolutional Neural Network
title_full_unstemmed An Efficient Dehazing Algorithm Based on the Fusion of Transformer and Convolutional Neural Network
title_short An Efficient Dehazing Algorithm Based on the Fusion of Transformer and Convolutional Neural Network
title_sort efficient dehazing algorithm based on the fusion of transformer and convolutional neural network
topic image dehazing
deep learning
convolutional neural network
Transformer
Transformer–Convolution fusion dehazing network (TCFDN)
url https://www.mdpi.com/1424-8220/23/1/43
work_keys_str_mv AT junxu anefficientdehazingalgorithmbasedonthefusionoftransformerandconvolutionalneuralnetwork
AT zixuanchen anefficientdehazingalgorithmbasedonthefusionoftransformerandconvolutionalneuralnetwork
AT haoluo anefficientdehazingalgorithmbasedonthefusionoftransformerandconvolutionalneuralnetwork
AT zheminglu anefficientdehazingalgorithmbasedonthefusionoftransformerandconvolutionalneuralnetwork
AT junxu efficientdehazingalgorithmbasedonthefusionoftransformerandconvolutionalneuralnetwork
AT zixuanchen efficientdehazingalgorithmbasedonthefusionoftransformerandconvolutionalneuralnetwork
AT haoluo efficientdehazingalgorithmbasedonthefusionoftransformerandconvolutionalneuralnetwork
AT zheminglu efficientdehazingalgorithmbasedonthefusionoftransformerandconvolutionalneuralnetwork