Learning Dual Convolutional Dictionaries for Image De-raining

被引:1
|
作者
Ge, Chengjie [1 ]
Fu, Xueyang [1 ]
Zha, Zheng-Jun [1 ]
机构
[1] Univ Sci & Technol China, Hefei, Anhui, Peoples R China
关键词
Image De-raining; Convolutional Dictionary Learning; Optimization; Deep Unfolding; Image Restoration; ALGORITHM;
D O I
10.1145/3503161.3548117
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Rain removal is a vital and highly ill-posed low-level vision task. While currently existing deep convolutional neural networks (CNNs) based image de-raining methods have achieved remarkable results, they still possess apparent shortcomings: First, most of the CNNs based models are lack of interpretability. Second, these models are not embedded with physical structures of rain streaks and background images. Third, they omit useful information in the background images. These deficiencies result in unsatisfied de-raining results in some sophisticated scenarios. To solve the above problems, we propose a Deep Dual Convolutional Dictionary Learning Network (DDCDNet) for these specific tasks. We firstly propose a new dual dictionary learning objective function, and then unfold it into the form of neural networks to learn prior knowledge from the data automatically. This network tries to learn the rain-streaks layer and the clean background using two dictionary learning networks instead of merely predicting the rain-streaks layer like most of the de-raining methods. To further increase the interpretability and generalization capability, we add sparsity and adaptive dictionary to our network to generate dynamic dictionary for each image based on content. Experimental results reveal that our model possesses outstanding de-raining ability on both synthetic and real-world data sets in terms of PSNR and SSIM as well as visual appearance.
引用
收藏
页码:6636 / 6644
页数:9
相关论文
共 50 条
  • [1] Successive Graph Convolutional Network for Image De-raining
    Xueyang Fu
    Qi Qi
    Zheng-Jun Zha
    Xinghao Ding
    Feng Wu
    John Paisley
    International Journal of Computer Vision, 2021, 129 : 1691 - 1711
  • [2] Image De-raining via Continual Learning
    Zhou, Man
    Xiao, Jie
    Chang, Yifan
    Fu, Xueyang
    Liu, Aiping
    Pan, Jinshan
    Zha, Zheng-Jun
    2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 4905 - 4914
  • [3] Successive Graph Convolutional Network for Image De-raining
    Fu, Xueyang
    Qi, Qi
    Zha, Zheng-Jun
    Ding, Xinghao
    Wu, Feng
    Paisley, John
    INTERNATIONAL JOURNAL OF COMPUTER VISION, 2021, 129 (05) : 1691 - 1711
  • [4] Image De-Raining Transformer
    Xiao, Jie
    Fu, Xueyang
    Liu, Aiping
    Wu, Feng
    Zha, Zheng-Jun
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (11) : 12978 - 12995
  • [5] Morphological networks for image de-raining
    Mondal, Ranjan
    Purkait, Pulak
    Santra, Sanchayan
    Chanda, Bhabatosh
    Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), 2019, 11414 LNCS : 262 - 275
  • [6] Morphological Networks for Image De-raining
    Mondal, Ranjan
    Purkait, Pulak
    Santra, Sanchayan
    Chanda, Bhabatosh
    DISCRETE GEOMETRY FOR COMPUTER IMAGERY, DGCI 2019, 2019, 11414 : 262 - 275
  • [7] IMAGE DE-RAINING VIA RDL: WHEN REWEIGHTED CONVOLUTIONAL SPARSE CODING MEETS DEEP LEARNING
    He, Jingwei
    Yu, Lei
    Yang, Wen
    2020 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH, AND SIGNAL PROCESSING, 2020, : 2548 - 2552
  • [8] Gradual Network for Single Image De-raining
    Yu, Weijiang
    Huang, Zhe
    Zhang, Wayne
    Feng, Litong
    Xiao, Nong
    PROCEEDINGS OF THE 27TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA (MM'19), 2019, : 1795 - 1804
  • [9] Disentangled Representation Learning and Enhancement Network for Single Image De-Raining
    Wang, Guoqing
    Sun, Changming
    Xu, Xing
    Li, Jingjing
    Wang, Zheng
    Ma, Zeyu
    PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2021, 2021, : 3015 - 3023
  • [10] DTDN: Dual-task De-raining Network
    Wang, Zheng
    Li, Jianwu
    Song, Ge
    PROCEEDINGS OF THE 27TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA (MM'19), 2019, : 1833 - 1841