Single image deraining via nonlocal squeeze-and-excitation enhancing network

被引:10
|
作者
Wang, Cong [1 ]
Fan, Wanshu [1 ]
Zhu, Honghe [1 ]
Su, Zhixun [1 ,2 ]
机构
[1] Dalian Univ Technol, Dalian, Peoples R China
[2] Guilin Univ Elect Technol, Guilin, Peoples R China
关键词
Single image de-raining; Convolutional Neural Network (CNN); Squeeze-and-excitation; Non-local mean; Dense network; RAIN; REMOVAL;
D O I
10.1007/s10489-020-01693-5
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Raindrop blur or rain streaks can severely degrade the visual quality of the images, which causes many practical vision systems to fail to work, such as autonomous driving and video surveillance. Hence, it is important to address the problem of single image de-raining. In this paper, we propose a novel deep network for single image de-raining. The proposed network consists of three stages, including encoder stage, Dense Non-Local Residual Block (DNLRB) stage, and decoder stage. As spatial contextual information has been analyzed to be meaningful for image de-raining (Huang et al. ??), we adopt squeeze-and-excitation enhancing on feature maps in each convolution layer for capturing spatial contextual information. In addition, to better leverage spatial contextual information for extracting rain components, the non-local mean operation has been embed in DNLRB. Both quantitative and qualitative experimental results demonstrate the proposed method performs favorably against the state-of-the-art de-raining methods. The source codes will be available at .
引用
收藏
页码:2932 / 2944
页数:13
相关论文
共 50 条
  • [11] Spatial-Spectral Squeeze-and-Excitation Residual Network for Hyperspectral Image Classification
    Wang, Li
    Peng, Jiangtao
    Sun, Weiwei
    [J]. REMOTE SENSING, 2019, 11 (07)
  • [12] Specific Emitter Identification via Squeeze-and-Excitation Neural Network in Frequency Domain
    Peng, Yinan
    Zhou, Yuan
    [J]. 2021 PROCEEDINGS OF THE 40TH CHINESE CONTROL CONFERENCE (CCC), 2021, : 8310 - 8314
  • [13] SQUEEZE-AND-EXCITATION WIDE RESIDUAL NETWORKS IN IMAGE CLASSIFICATION
    Zhong, Xian
    Gong, Oubo
    Huang, Wenxin
    Li, Lin
    Xia, Hongxia
    [J]. 2019 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2019, : 395 - 399
  • [14] FuSENet: fused squeeze-and-excitation network for spectral-spatial hyperspectral image classification
    Roy, Swalpa Kumar
    Dubey, Shiv Ram
    Chatterjee, Subhrasankar
    Baran Chaudhuri, Bidyut
    [J]. IET IMAGE PROCESSING, 2020, 14 (08) : 1653 - 1661
  • [15] Denoising real bursts with squeeze-and-excitation residual network
    Tan, Hanlin
    Xiao, Huaxin
    Lai, Shiming
    Liu, Yu
    Zhang, Maojun
    [J]. IET IMAGE PROCESSING, 2020, 14 (13) : 3095 - 3104
  • [16] Convolutional squeeze-and-excitation network for ECG arrhythmia detection
    Ge, Rongjun
    Shen, Tengfei
    Zhou, Ying
    Liu, Chengyu
    Zhang, Libo
    Yang, Benqiang
    Yan, Ying
    Coatrieux, Jean-Louis
    Chen, Yang
    [J]. ARTIFICIAL INTELLIGENCE IN MEDICINE, 2021, 121
  • [17] DSSESKN: A depthwise separable squeeze-and-excitation selective kernel network for art image classification
    Zhang, Shaojie
    [J]. EAI ENDORSED TRANSACTIONS ON SCALABLE INFORMATION SYSTEMS, 2022, 9 (36)
  • [18] Coupled Squeeze-and-Excitation Blocks Based CNN for Image Compression
    Du, Jing
    Xu, Yang
    Wei, Zhihui
    [J]. INTELLIGENCE SCIENCE AND BIG DATA ENGINEERING: VISUAL DATA ENGINEERING, PT I, 2019, 11935 : 201 - 212
  • [19] Single Image Deraining via Recurrent Hierarchy Enhancement Network
    Yang, Youzhao
    Lu, Hong
    [J]. PROCEEDINGS OF THE 27TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA (MM'19), 2019, : 1814 - 1822
  • [20] Siamese Network Object Tracking Algorithm Based on Squeeze-and-Excitation
    Wang, Jianwen
    Li, Aimin
    Liu, Teng
    [J]. 2020 IEEE INTERNATIONAL CONFERENCE ON SYSTEMS, MAN, AND CYBERNETICS (SMC), 2020, : 3582 - 3587