Dual attention residual group networks for single image deraining

被引:7
作者
Zhang, Hai [1 ]
Xie, Qiangqiang [1 ]
Lu, Bei [1 ]
Gai, Shan [1 ]
机构
[1] Nanchang Hangkong Univ, Sch Informat Engn, Nanchang 330063, Jiangxi, Peoples R China
关键词
Spatial attention; Residual groups; Channel attention; Single image rain removal; VISION; RAIN; REMOVAL;
D O I
10.1016/j.dsp.2021.103106
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Single image deraining is one of challenges in image processing. An efficient algorithm for single image deraining can significantly improve the image quality in severe weather conditions. Existing deraining algorithms only pay attention to spatial characteristics or channel information, which leads to poor performance of the network. In this paper, we propose a novel dual attention residual group network (DARGNet) to get better deraining performance. Specifically, the framework of dual attention includes spatial attention and channel attention. The spatial attention can extract the multi-scale feature to adapt to different shapes and size of the rain streaks. Meanwhile, channel attention has established the dependence relationship among different channels. In addition, in order to simplify the structure, we integrate the dual attention module and convolution layers into the residual groups, which also improves information transmission. Extensive experiments on synthesized and real-world datasets demonstrate that the proposed network achieves a good effect of deraining tasks. The source code is available at https://github.com/zhanghai404. (C) 2021 Elsevier Inc. All rights reserved.
引用
收藏
页数:11
相关论文
共 40 条
  • [1] Transformed Low-rank Model for Line Pattern Noise Removal
    Chang, Yi
    Yan, Luxin
    Zhong, Sheng
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 1735 - 1743
  • [2] SCA-CNN: Spatial and Channel-wise Attention in Convolutional Networks for Image Captioning
    Chen, Long
    Zhang, Hanwang
    Xiao, Jun
    Nie, Liqiang
    Shao, Jian
    Liu, Wei
    Chua, Tat-Seng
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 6298 - 6306
  • [3] A Generalized Low-Rank Appearance Model for Spatio-Temporally Correlated Rain Streaks
    Chen, Yi-Lei
    Hsu, Chiou-Ting
    [J]. 2013 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2013, : 1968 - 1975
  • [4] Xception: Deep Learning with Depthwise Separable Convolutions
    Chollet, Francois
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 1800 - 1807
  • [5] Histograms of oriented gradients for human detection
    Dalal, N
    Triggs, B
    [J]. 2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, : 886 - 893
  • [6] Lightweight Pyramid Networks for Image Deraining
    Fu, Xueyang
    Liang, Borong
    Huang, Yue
    Ding, Xinghao
    Paisley, John
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2020, 31 (06) : 1794 - 1807
  • [7] Removing rain from single images via a deep detail network
    Fu, Xueyang
    Huang, Jiabin
    Zeng, Delu
    Huang, Yue
    Ding, Xinghao
    Paisley, John
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 1715 - 1723
  • [8] Clearing the Skies: A Deep Network Architecture for Single-Image Rain Removal
    Fu, Xueyang
    Huang, Jiabin
    Ding, Xinghao
    Liao, Yinghao
    Paisley, John
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2017, 26 (06) : 2944 - 2956
  • [9] Speckle noise reduction in medical ultrasound image using monogenic wavelet and Laplace mixture distribution
    Gai, Shan
    Zhang, Boyu
    Yang, Cihui
    Yu, Lei
    [J]. DIGITAL SIGNAL PROCESSING, 2018, 72 : 192 - 207
  • [10] Vision and rain
    Garg, Kshitiz
    Nayar, Shree K.
    [J]. INTERNATIONAL JOURNAL OF COMPUTER VISION, 2007, 75 (01) : 3 - 27