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

被引:3
作者
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
    Bossu, Jeremie
    Hautiere, Nicolas
    Tarel, Jean-Philippe
    [J]. INTERNATIONAL JOURNAL OF COMPUTER VISION, 2011, 93 (03) : 348 - 367
  • [2] Gated Context Aggregation Network for Image Dehazing and Deraining
    Chen, Dongdong
    He, Mingming
    Fan, Qingnan
    Liao, Jing
    Zhang, Liheng
    Hou, Dongdong
    Yuan, Lu
    Hua, Gang
    [J]. 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
    Chen, Jie
    Tan, Cheen-Hau
    Hou, Junhui
    Chau, Lap-Pui
    Li, He
    [J]. 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
    Chen, Yi-Lei
    Hsu, Chiou-Ting
    [J]. 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
    Fu, Xueyang
    Huang, Jiabin
    Ding, Xinghao
    Liao, Yinghao
    Paisley, John
    [J]. 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
    Guo, Mengyan
    Huang, Bo
    Zhang, Juan
    Wang, Feng
    Zhang, Yan
    Fang, Zhijun
    [J]. 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
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. COMPUTER VISION - ECCV 2016, PT IV, 2016, 9908 : 630 - 645