Coarse-to-fine multiscale fusion network for single image deraining

被引:4
作者
Zhang, Jiahao [1 ]
Zhang, Juan [1 ]
Wu, Xing [2 ]
Shi, Zhicai [3 ]
Hwang, Jenq-Neng [4 ]
机构
[1] Shanghai Univ Engn Sci, Sch Elect & Elect Engn, Joint Int Lab Intelligent Percept & Control, Shanghai, Peoples R China
[2] Shanghai Univ, Dept Comp Sci & Technol, Shanghai, Peoples R China
[3] Shanghai Key Lab Integrated Adm Technol Informat, Shanghai, Peoples R China
[4] Univ Washington, Dept Elect & Comp Engn, Seattle, WA 98195 USA
基金
中国国家自然科学基金;
关键词
rain removal; multiscale information exchange; feature attention compensation; image degradations; REMOVAL; RAIN;
D O I
10.1117/1.JEI.31.4.043003
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Cameras convert optical signals into electronic information to enable image acquisition. However, cameras are susceptible to environmental factors such as light and weather conditions. The information captured by cameras in rainy condition tends to degrade, resulting in reduced contrast and visibility of the images. Deraining aims to restore the image degraded by raindrops and rain accumulation in rainy weather. Most state-of-the-art, end-to-end neural network-based deraining methods have achieved satisfactory results. By considering the features from different scales, we proposed a further improved multiscale information exchange module to extract features from different scales. As a result, the structure achieves a competitive effect on rain mask detection. Moreover, experiments have found that the rain layer-based prediction models tend to leave rain streak residuals in specific image patches. Therefore, we further developed the Feature Attention Compensation module (FAC) to better utilize the deraining image and rain layer information obtained from the multiscale model, which is demonstrated to boost deraining performance. To summarize, we design a coarse-to-fine rain removal model, starting with a rain detection network to get a coarse rain-free image, which is further refined with the FAC module to eventually create a fine rain-free image. We conduct our experiments on both synthetic and real-world datasets. Quantitative and qualitative experimental results demonstrate that the proposed method outperforms the state-of-the-art deraining methods. Source code will be available at
引用
收藏
页数:15
相关论文
共 42 条
[1]   Rain or Snow Detection in Image Sequences Through Use of a Histogram of Orientation of Streaks [J].
Bossu, Jeremie ;
Hautiere, Nicolas ;
Tarel, Jean-Philippe .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2011, 93 (03) :348-367
[2]   Gated Context Aggregation Network for Image Dehazing and Deraining [J].
Chen, Dongdong ;
He, Mingming ;
Fan, Qingnan ;
Liao, Jing ;
Zhang, Liheng ;
Hou, Dongdong ;
Yuan, Lu ;
Hua, Gang .
2019 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV), 2019, :1375-1383
[3]   Robust Video Content Alignment and Compensation for Rain Removal in a CNN Framework [J].
Chen, Jie ;
Tan, Cheen-Hau ;
Hou, Junhui ;
Chau, Lap-Pui ;
Li, He .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :6286-6295
[4]   A Generalized Low-Rank Appearance Model for Spatio-Temporally Correlated Rain Streaks [J].
Chen, Yi-Lei ;
Hsu, Chiou-Ting .
2013 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2013, :1968-1975
[5]  
Feng RC, 2020, Arxiv, DOI arXiv:2008.00239
[6]   Clearing the Skies: A Deep Network Architecture for Single-Image Rain Removal [J].
Fu, Xueyang ;
Huang, Jiabin ;
Ding, Xinghao ;
Liao, Yinghao ;
Paisley, John .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2017, 26 (06) :2944-2956
[7]  
Garg K, 2004, PROC CVPR IEEE, P528
[8]   DFBDehazeNet: an end-to-end dense feedback network for single image dehazing [J].
Guo, Mengyan ;
Huang, Bo ;
Zhang, Juan ;
Wang, Feng ;
Zhang, Yan ;
Fang, Zhijun .
JOURNAL OF ELECTRONIC IMAGING, 2021, 30 (03)
[9]  
He KM, 2020, IEEE T PATTERN ANAL, V42, P386, DOI [10.1109/TPAMI.2018.2844175, 10.1109/ICCV.2017.322]
[10]   Identity Mappings in Deep Residual Networks [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
COMPUTER VISION - ECCV 2016, PT IV, 2016, 9908 :630-645