Sequential Dual Attention Network for Rain Streak Removal in a Single Image

Chih Yang Lin, Zhuang Tao, Ai Sheng Xu, Li Wei Kang*, Fityanul Akhyar

*此作品的通信作者

研究成果: 雜誌貢獻期刊論文同行評審

5 引文 斯高帕斯(Scopus)

摘要

Various weather conditions, such as rain, haze, or snow, can degrade visual quality in images/videos, which may significantly degrade the performance of related applications. In this paper, a novel framework based on sequential dual attention deep network is proposed for removing rain streaks (deraining) in a single image, called by SSDRNet (Sequential dual attention-based Single image DeRaining deep Network). Since the inherent correlation among rain steaks within an image should be stronger than that between the rain streaks and the background (non-rain) pixels, a two-stage learning strategy is implemented to better capture the distribution of rain streaks within a rainy image. The two-stage deep neural network primarily involves three blocks: residual dense blocks (RDBs), sequential dual attention blocks (SDABs), and multi-scale feature aggregation modules (MAMs), which are all delicately and specifically designed for rain removal. The two-stage strategy successfully learns very fine details of the rain steaks of the image and then clearly removes them. Extensive experimental results have shown that the proposed deep framework achieves the best performance on qualitative and quantitative metrics compared with state-of-the-art methods. The corresponding code and the trained model of the proposed SSDRNet have been available online at https://github.com/fityanul/SDAN-for-Rain-Removal.

原文英語
文章編號9206069
頁(從 - 到)9250-9265
頁數16
期刊IEEE Transactions on Image Processing
29
DOIs
出版狀態已發佈 - 2020

ASJC Scopus subject areas

  • 軟體
  • 電腦繪圖與電腦輔助設計

指紋

深入研究「Sequential Dual Attention Network for Rain Streak Removal in a Single Image」主題。共同形成了獨特的指紋。

引用此