Lightweight Pyramid Networks for Image Deraining

被引:304
作者
Fu, Xueyang [1 ,2 ]
Liang, Borong [1 ]
Huang, Yue [1 ]
Ding, Xinghao [1 ]
Paisley, John [3 ,4 ]
机构
[1] Xiamen Univ, Fujian Key Lab Sensing & Comp Smart City, Sch Informat Sci & Engn, Xiamen 361005, Peoples R China
[2] Univ Sci & Technol China, Sch Informat Sci & Technol, Hefei 230026, Peoples R China
[3] Columbia Univ, Dept Elect Engn, New York, NY 10027 USA
[4] Columbia Univ, Data Sci Inst, New York, NY 10027 USA
基金
中国国家自然科学基金;
关键词
Rain; Laplace equations; Feature extraction; Learning systems; Task analysis; Knowledge engineering; Computer vision; Deep convolutional neural network (CNN); image pyramid; lightweight networks; rain removal; residual learning; QUALITY ASSESSMENT; RAIN; CLASSIFICATION;
D O I
10.1109/TNNLS.2019.2926481
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Existing deep convolutional neural networks (CNNs) have found major success in image deraining, but at the expense of an enormous number of parameters. This limits their potential applications, e.g., in mobile devices. In this paper, we propose a lightweight pyramid networt (LPNet) for single-image deraining. Instead of designing a complex network structure, we use domain-specific knowledge to simplify the learning process. In particular, we find that by introducing the mature Gaussian-Laplacian image pyramid decomposition technology to the neural network, the learning problem at each pyramid level is greatly simplified and can be handled by a relatively shallow network with few parameters. We adopt recursive and residual network structures to build the proposed LPNet, which has less than 8K parameters while still achieving the state-of-the-art performance on rain removal. We also discuss the potential value of LPNet for other low- and high-level vision tasks.
引用
收藏
页码:1794 / 1807
页数:14
相关论文
共 54 条
[1]  
[Anonymous], puter Vision and Pattern Recognition
[2]  
[Anonymous], C COMP VIS PATT REC
[3]  
[Anonymous], 2017, ARXIV170402071
[4]  
[Anonymous], IEEE T PATTERN ANAL
[5]  
[Anonymous], 2004, P IEEE COMPUT VIS PA
[6]  
[Anonymous], 2017, P IEEE C COMP VIS PA
[7]  
[Anonymous], 2017, P IEEE C COMP VIS PA
[8]   Contour Detection and Hierarchical Image Segmentation [J].
Arbelaez, Pablo ;
Maire, Michael ;
Fowlkes, Charless ;
Malik, Jitendra .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2011, 33 (05) :898-916
[9]   On the Landau-gauge adjoint quark propagator [J].
August, Daniel ;
Maas, Axel .
JOURNAL OF HIGH ENERGY PHYSICS, 2013, (07)
[10]   Analysis of Rain and Snow in Frequency Space [J].
Barnum, Peter C. ;
Narasimhan, Srinivasa ;
Kanade, Takeo .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2010, 86 (2-3) :256-274