Unsupervised Network for Single Image Raindrop Removal

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Huijiao, Zhao, Shenghao, Yu, Lei, Yang, Xulei
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909415315079168
author Wang, Huijiao
Zhao, Shenghao
Yu, Lei
Yang, Xulei
author_facet Wang, Huijiao
Zhao, Shenghao
Yu, Lei
Yang, Xulei
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2412_03019
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Unsupervised Network for Single Image Raindrop Removal
Wang, Huijiao
Zhao, Shenghao
Yu, Lei
Yang, Xulei
Computer Vision and Pattern Recognition
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.
title Unsupervised Network for Single Image Raindrop Removal
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.03019