arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1808.08545cs.CV
收藏

基于核函数引导的CNN单图像雨带去除

Rain Streak Removal for Single Image via Kernel Guided CNN

University of Electronic Science and Technology of China(电子科技大学);Huazhong University of Science and Technology(华中科技大学)

Ye-Tao Wang, Xi-Le Zhao, Tai-Xiang Jiang, Liang-Jian Deng, Yi Chang, Ting-Zhu Huang

摘要:

雨痕去除是一个重要的问题,最近已被广泛研究。现有的方法,特别是新出现的深度学习方法,在许多情况下可以很好地去除雨纹。然而,雨条形成过程中的关键因素,导致线图案外观的运动模糊被深度学习雨条纹方法忽略,这导致过度偏离或欠偏离的结果。在本文中,我们提出了一种新的雨条纹去除方法,使用核引导的卷积神经网络(KGCNN),实现了最先进的性能与简单的网络架构。我们首先建立了雨条纹干涉的运动模糊模型。然后,我们的框架开始学习的运动模糊内核,这是由两个因素,包括角度和长度,由一个普通的神经网络,表示为参数网络,从一个补丁的纹理组件。然后,在维度拉伸操作之后,学习的运动模糊核被拉伸成与雨片具有相同空间大小的退化图。拉伸的退化图与纹理补丁一起随后被输入到一个derain卷积网络中,这是一个典型的ResNet架构,并在学习的运动模糊内核的指导下训练输出雨纹。在大量的合成数据和真实数据上进行的实验表明,该方法在去除雨纹的同时,保持了纹理和对比度。

英文摘要:

Rain streak removal is an important issue and has recently been investigated extensively. Existing methods, especially the newly emerged deep learning methods, could remove the rain streaks well in many cases. However the essential factor in the generative procedure of the rain streaks, i.e., the motion blur, which leads to the line pattern appearances, were neglected by the deep learning rain streaks approaches and this resulted in over-derain or under-derain results. In this paper, we propose a novel rain streak removal approach using a kernel guided convolutional neural network (KGCNN), achieving the state-of-the-art performance with simple network architectures. We first model the rain streak interference with its motion blur mechanism. Then, our framework starts with learning the motion blur kernel, which is determined by two factors including angle and length, by a plain neural network, denoted as parameter net, from a patch of the texture component. Then, after a dimensionality stretching operation, the learned motion blur kernel is stretched into a degradation map with the same spatial size as the rainy patch. The stretched degradation map together with the texture patch is subsequently input into a derain convolutional network, which is a typical ResNet architecture and trained to output the rain streaks with the guidance of the learned motion blur kernel. Experiments conducted on extensive synthetic and real data demonstrate the effectiveness of the proposed method, which preserves the texture and the contrast while removing the rain streaks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑