arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.03019cs.CV

用于单幅图像雨滴去除的无监督网络

Unsupervised Network for Single Image Raindrop Removal

Wuhan University(武汉大学);National University of Singapore (NUS)(新加坡国立大学);A*STAR(新加坡科技研究局)

Huijiao Wang, Shenghao Zhao, Lei Yu, Xulei Yang

AI总结:

针对监督式雨滴去除依赖难获取成对图像的问题,提出基于循环架构与反馈机制的无监督网络,仅用非配对图像即可分离雨滴并恢复干净背景。

摘要:

由雨滴造成的图像质量下降是降低视觉系统性能的最重要且最具挑战性的问题之一。现有大多数雨滴去除算法基于使用成对图像的监督学习方法,而成对图像在真实世界应用中很难获取。本研究提出了一种基于无监督学习的雨滴去除深度神经网络,该网络仅需要两个分别包含带雨滴图像和无雨滴图像的非配对图像集。我们提出的模型基于循环网络架构执行层分离,旨在将雨天图像分离为雨滴层、透明度掩码和干净背景层。干净背景层是目标雨滴去除结果,而透明度掩码指示雨滴的空间位置。此外,所提出的模型应用反馈机制,通过利用高层信息细化低层表示来促进层分离,即前一次迭代的输出与带雨滴的输入图像一起被用作下一次迭代的输入。因此,雨滴可以通过这种反馈方式被逐步去除。在雨滴基准数据集上进行的大量实验,在定量指标和视觉质量方面证明了所提方法的有效性。

英文摘要:

Image quality degradation caused by raindrops is one of the most important but challenging problems that reduce the performance of vision systems. Most existing raindrop removal algorithms are based on a supervised learning method using pairwise images, which are hard to obtain in real-world applications. This study proposes a deep neural network for raindrop removal based on unsupervised learning, which only requires two unpaired image sets with and without raindrops. Our proposed model performs layer separation based on cycle network architecture, which aims to separate a rainy image into a raindrop layer, a transparency mask, and a clean background layer. The clean background layer is the target raindrop removal result, while the transparency mask indicates the spatial locations of the raindrops. In addition, the proposed model applies a feedback mechanism to benefit layer separation by refining low-level representation with high-level information. i.e., the output of the previous iteration is used as input for the next iteration, together with the input image with raindrops. As a result, raindrops could be gradually removed through this feedback manner. Extensive experiments on raindrop benchmark datasets demonstrate the effectiveness of the proposed method on quantitative metrics and visual quality.

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

↑