Mar, 2024

TRG-Net:一种可解释和可控的雨生成器

TL;DR探索和建模降雨生成机制对于增加配对数据减轻降雨图像处理模型的训练任务至关重要,该研究提出了一种新颖的基于深度学习的降雨生成器,充分考虑了降雨背后的物理生成机制,并将基本降雨因素(即形状、方向、长度、宽度和稀疏度)的学习明确地编码到深度网络中。我们首次通过合理采用滤波器参数化技术实现了一个在降雨因素上可精细控制且能够纯粹通过数据学习这些因素的深度网络。我们的生成实验表明,由提出的降雨生成器生成的降雨不仅质量更高,而且对于去雨和下游任务比目前最先进的降雨生成方法更有效。此外,我们的配对数据增强实验,包括分布内和分布外(OOD),进一步验证了我们模型生成的样本的多样性,用于分布内去雨和 OOD 泛化任务。