TRG-Net: An Interpretable and Controllable Rain Generator

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Pang, Zhiqiang, Wang, Hong, Xie, Qi, Meng, Deyu, Xu, Zongben
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910426115080192
author Pang, Zhiqiang
Wang, Hong
Xie, Qi
Meng, Deyu
Xu, Zongben
author_facet Pang, Zhiqiang
Wang, Hong
Xie, Qi
Meng, Deyu
Xu, Zongben
contents Exploring and modeling rain generation mechanism is critical for augmenting paired data to ease training of rainy image processing models. Against this task, this study proposes a novel deep learning based rain generator, which fully takes the physical generation mechanism underlying rains into consideration and well encodes the learning of the fundamental rain factors (i.e., shape, orientation, length, width and sparsity) explicitly into the deep network. Its significance lies in that the generator not only elaborately design essential elements of the rain to simulate expected rains, like conventional artificial strategies, but also finely adapt to complicated and diverse practical rainy images, like deep learning methods. By rationally adopting filter parameterization technique, we first time achieve a deep network that is finely controllable with respect to rain factors and able to learn the distribution of these factors purely from data. Our unpaired generation experiments demonstrate that the rain generated by the proposed rain generator is not only of higher quality, but also more effective for deraining and downstream tasks compared to current state-of-the-art rain generation methods. Besides, the paired data augmentation experiments, including both in-distribution and out-of-distribution (OOD), further validate the diversity of samples generated by our model for in-distribution deraining and OOD generalization tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2403_09993
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TRG-Net: An Interpretable and Controllable Rain Generator
Pang, Zhiqiang
Wang, Hong
Xie, Qi
Meng, Deyu
Xu, Zongben
Computer Vision and Pattern Recognition
Image and Video Processing
Exploring and modeling rain generation mechanism is critical for augmenting paired data to ease training of rainy image processing models. Against this task, this study proposes a novel deep learning based rain generator, which fully takes the physical generation mechanism underlying rains into consideration and well encodes the learning of the fundamental rain factors (i.e., shape, orientation, length, width and sparsity) explicitly into the deep network. Its significance lies in that the generator not only elaborately design essential elements of the rain to simulate expected rains, like conventional artificial strategies, but also finely adapt to complicated and diverse practical rainy images, like deep learning methods. By rationally adopting filter parameterization technique, we first time achieve a deep network that is finely controllable with respect to rain factors and able to learn the distribution of these factors purely from data. Our unpaired generation experiments demonstrate that the rain generated by the proposed rain generator is not only of higher quality, but also more effective for deraining and downstream tasks compared to current state-of-the-art rain generation methods. Besides, the paired data augmentation experiments, including both in-distribution and out-of-distribution (OOD), further validate the diversity of samples generated by our model for in-distribution deraining and OOD generalization tasks.
title TRG-Net: An Interpretable and Controllable Rain Generator
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2403.09993