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

基于透过率和大气光估计的雨景单图像深度去噪

Deep Single Image Deraining Via Estimating Transmission and Atmospheric Light in rainy Scenes

University of Electronic Science and Technology of China(电子科技大学);The University of Adelaide(阿德莱德大学);Shanghai Jiao Tong University(上海交通大学);Beijing Jiao Tong University(北京交通大学)

Yinglong Wang, Qinfeng Shi, Ehsan Abbasnejad, Chao Ma, Xiaoping Ma, Bing Zeng

摘要:

图像/视频中的雨水去除仍然是计算机视觉领域的一个重要课题,受到越来越多人的关注。传统的方法通常采用不完全先验或滤波器(如引导滤波器)来消除降雨影响。深度学习提供了更多的概率来更好地解决这个任务。然而,它们要么直接从雨天图像中评估背景,要么先学习一个雨残差,然后减去残差,以获得清晰的背景。在基于深度学习的去雨方法中没有使用其他模型来去除雨水并获得关于下雨场景的其他信息。本文利用一种广泛应用的图像退化模型,从大气散射原理出发,对雨天图像的形成过程进行建模,并尝试了解雨天场景中大气光的传输情况,进而去除雨水。为了达到这一目标,我们提出了一个鲁棒的评估方法,全球大气光在下雨的场景。我们不是直接使用估计的大气光来学习网络来计算传输,而是利用它作为地面实况,并设计了一个简单但新颖的三角形网络结构来学习每一幅雨天图像的大气光,然后在传输网络的训练过程中对网络进行微调,以获得更好的大气光估计。在传输网络中利用更高效的ShuffleNet单元学习传输映射,并通过图像退化模型得到去雨图像。通过主观和客观的比较,我们的方法优于选定的国家的最先进的作品。

英文摘要:

Rain removal in images/videos is still an important task in computer vision field and attracting attentions of more and more people. Traditional methods always utilize some incomplete priors or filters (e.g. guided filter) to remove rain effect. Deep learning gives more probabilities to better solve this task. However, they remove rain either by evaluating background from rainy image directly or learning a rain residual first then subtracting the residual to obtain a clear background. No other models are used in deep learning based de-raining methods to remove rain and obtain other information about rainy scenes. In this paper, we utilize an extensively-used image degradation model which is derived from atmospheric scattering principles to model the formation of rainy images and try to learn the transmission, atmospheric light in rainy scenes and remove rain further. To reach this goal, we propose a robust evaluation method of global atmospheric light in a rainy scene. Instead of using the estimated atmospheric light directly to learn a network to calculate transmission, we utilize it as ground truth and design a simple but novel triangle-shaped network structure to learn atmospheric light for every rainy image, then fine-tune the network to obtain a better estimation of atmospheric light during the training of transmission network. Furthermore, more efficient ShuffleNet Units are utilized in transmission network to learn transmission map and the de-raining image is then obtained by the image degradation model. By subjective and objective comparisons, our method outperforms the selected state-of-the-art works.

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

↑