Single Image Defogging Algorithm Based on Attention Mechanism

被引:2
|
作者
Cao Ruihu [1 ]
Zhang Pengchao [1 ,2 ]
Wang Lei [1 ]
Zhang Fan [1 ]
Kang Jie [1 ]
机构
[1] Shaanxi Univ Technol, Sch Mech Engn, Hanzhong 723000, Shaanxi, Peoples R China
[2] Shaanxi Key Lab Ind Automat, Hanzhong 723000, Shaanxi, Peoples R China
关键词
image processing; image defogging; residual network; attention mechanism;
D O I
10.3788/LOP213235
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Present image-defogging methods have a range of problems: insufficient numbers of real datasets, local contrast imbalance, and defogging image distortion. This paper proposes a novel defogging network model (Densely Resnet with SKattention-Dehaze Net, DRS-Dehaze Net) that mitigates defogging image distortion. First, the fogged image is transformed into a multi-angle feature input map by the preprocessing module. The feature information is then extracted and redistributed through a dense residual architecture with an attention mechanism. Finally, the features are fused to output a fog-free image. Experimental comparison results confirmed a better defogging effect of the proposed algorithm than that of other algorithms. Our model effectively improves the distortion in defogged images and enhances the image clarity to a certain extent.
引用
收藏
页数:7
相关论文
共 21 条
  • [1] Baomar H, 2016, PROCEEDINGS OF 2016 IEEE SYMPOSIUM SERIES ON COMPUTATIONAL INTELLIGENCE (SSCI)
  • [2] DehazeNet: An End-to-End System for Single Image Haze Removal
    Cai, Bolun
    Xu, Xiangmin
    Jia, Kui
    Qing, Chunmei
    Tao, Dacheng
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2016, 25 (11) : 5187 - 5198
  • [3] Mask-Wearing Detection Method Based on YOLO-Mask
    Cao Chengshuo
    Yuan Jie
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (08)
  • [4] Dehazing Using Color-Lines
    Fattal, Raanan
    [J]. ACM TRANSACTIONS ON GRAPHICS, 2014, 34 (01):
  • [5] Unsupervised Single Image Dehazing Using Dark Channel Prior Loss
    Golts, Alona
    Freedman, Daniel
    Elad, Michael
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 : 2692 - 2701
  • [6] Single Image Haze Removal Using Dark Channel Prior
    He, Kaiming
    Sun, Jian
    Tang, Xiaoou
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2011, 33 (12) : 2341 - 2353
  • [7] Benchmarking Single-Image Dehazing and Beyond
    Li, Boyi
    Ren, Wenqi
    Fu, Dengpan
    Tao, Dacheng
    Feng, Dan
    Zeng, Wenjun
    Wang, Zhangyang
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2019, 28 (01) : 492 - 505
  • [8] AOD-Net: All-in-One Dehazing Network
    Li, Boyi
    Peng, Xiulian
    Wang, Zhangyang
    Xu, Jizheng
    Feng, Dan
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 4780 - 4788
  • [9] You Only Look Yourself: Unsupervised and Untrained Single Image Dehazing Neural Network
    Li, Boyun
    Gou, Yuanbiao
    Gu, Shuhang
    Liu, Jerry Zitao
    Zhou, Joey Tianyi
    Peng, Xi
    [J]. INTERNATIONAL JOURNAL OF COMPUTER VISION, 2021, 129 (05) : 1754 - 1767
  • [10] Liu X., 2020, Open Access Libr. J, V7, P1, DOI [10.4236/oalib.1106280, DOI 10.4236/OALIB.1106280]