Image Dehazing Algorithm Based on Deep Learning Coupled Local and Global Features

被引:8
|
作者
Li, Shuping [1 ]
Yuan, Qianhao [1 ]
Zhang, Yeming [1 ,2 ,3 ]
Lv, Baozhan [1 ]
Wei, Feng [1 ]
机构
[1] Henan Polytech Univ, Sch Mech & Power Engn, Jiaozuo 454000, Henan, Peoples R China
[2] Zhejiang Univ, State Key Lab Fluid Power & Mechatron Syst, Hangzhou 310027, Peoples R China
[3] Beijing Key Lab Pneumat Thermal Energy Storage &, Beijing 100191, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2022年 / 12卷 / 17期
关键词
image dehazing; convolutional neural network; vision transformer; hybrid feature fusion;
D O I
10.3390/app12178552
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
To address the problems that most convolutional neural network-based image defogging algorithm models capture incomplete global feature information and incomplete defogging, this paper proposes an end-to-end convolutional neural network and vision transformer hybrid image defogging algorithm. First, the shallow features of the haze image were extracted by a preprocessing module. Then, a symmetric network structure including a convolutional neural network (CNN) branch and a vision transformer branch was used to capture the local features and global features of the haze image, respectively. The mixed features were fused using convolutional layers to cover the global representation while retaining the local features. Finally, the features obtained by the encoder and decoder were fused to obtain richer feature information. The experimental results show that the proposed defogging algorithm achieved better defogging results in both the uniform and non-uniform haze datasets, solves the problems of dark and distorted colors after image defogging, and the recovered images are more natural for detail processing.
引用
收藏
页数:14
相关论文
共 50 条
  • [1] Deep Learning based Single Image Dehazing
    Suarez, Patricia L.
    Sappa, Angel D.
    Vintimilla, Boris X.
    Hammoud, Riad, I
    [J]. PROCEEDINGS 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2018, : 1250 - 1257
  • [2] Image Dehazing Based on Local and Non-Local Features
    Jiao, Qingliang
    Liu, Ming
    Ning, Bu
    Zhao, Fengfeng
    Dong, Liquan
    Kong, Lingqin
    Hui, Mei
    Zhao, Yuejin
    [J]. FRACTAL AND FRACTIONAL, 2022, 6 (05)
  • [3] Multi-scale single image dehazing based on the fusion of global and local features
    Chen, Ziyu
    Zhuang, Huaiyu
    Han, Jia
    Cui, Yani
    Deng, Jiaxian
    [J]. IET IMAGE PROCESSING, 2022, 16 (08) : 2049 - 2062
  • [4] Image enhancement algorithm based on global and local multi features
    Liu Yu-jie
    Zhu Shao-ping
    [J]. CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2020, 35 (05) : 508 - 512
  • [5] A Comprehensive Survey on Image Dehazing Based on Deep Learning
    Gui, Jie
    Cong, Xiaofeng
    Cao, Yuan
    Ren, Wenqi
    Zhang, Jun
    Zhang, Jing
    Tao, Dacheng
    [J]. PROCEEDINGS OF THE THIRTIETH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, IJCAI 2021, 2021, : 4426 - 4433
  • [6] A Sensor Image Dehazing Algorithm Based on Feature Learning
    Liu, Kun
    He, Linyuan
    Ma, Shiping
    Gao, Shan
    Bi, Duyan
    [J]. SENSORS, 2018, 18 (08)
  • [7] Learning Deep Priors for Image Dehazing
    Liu, Yang
    Pan, Jinshan
    Ren, Jimmy
    Su, Zhixun
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 2492 - 2500
  • [8] Image visual saliency detection algorithm based on local and global features
    [J]. 1600, Centre for Environment Social and Economic Research, Post Box No. 113, Roorkee, 247667, India (51):
  • [9] Deep Learning approaches for Image Dehazing
    Pavethra, M.
    Umadevi, M.
    [J]. 6TH IEEE INTERNATIONAL CONFERENCE ON RECENT ADVANCES AND INNOVATIONS IN ENGINEERING (ICRAIE), 2021,
  • [10] Research Advances on Deep Learning Based Single Image Dehazing
    Jia, Tong-Yao
    Zhuo, Li
    Li, Jia-Feng
    Zhang, Jing
    [J]. Tien Tzu Hsueh Pao/Acta Electronica Sinica, 2023, 51 (01): : 231 - 245