Coupled Rain Streak and Background Estimation via Separable Element-Wise Attention

Yinjie Tan, Qiang Wen, Jing Qin, Jianbo Jiao, Guoqiang Han, Shengfeng He

Research output: Journal article publicationJournal articleAcademic researchpeer-review

Abstract

Single image de-raining is challenging especially in the scenarios with dense rain streaks. Existing methods resolve this problem by predicting the rain streaks of the image, which constrains the network to focus on local rain streaks features. However, dense rain streaks are visually similar to mist or fog (with large intensities), in this case, the training objective should be shifted to image recovery instead of extracting rain streaks. In this paper, we propose a coupled rain streak and background estimation network that explores the intrinsic relations between two tasks. In particular, our network produces task-dependent feature maps, each part of the features correspond to the estimation of rain streak and background. Furthermore, to inject element-wise attention to all the convolutional blocks for better understanding the rain streaks distribution, we propose a Separable Element-wise Attention mechanism. In this way, dense element-wise attention can be obtained by a sequence of channel and spatial attention modules, with negligible computation. Extensive experiments demonstrate that the proposed method outperforms state-of-the-arts on 5 existing synthesized rain datasets and the real-world scenarios, without extra multi-scale or recurrent structure.

Original languageEnglish
Article number8963735
Pages (from-to)16627-16636
Number of pages10
JournalIEEE Access
Volume8
DOIs
Publication statusPublished - Jan 2020

Keywords

  • Background estimation
  • de-raining
  • element-wise attention

ASJC Scopus subject areas

  • Computer Science(all)
  • Materials Science(all)
  • Engineering(all)

Fingerprint

Dive into the research topics of 'Coupled Rain Streak and Background Estimation via Separable Element-Wise Attention'. Together they form a unique fingerprint.

Cite this