北大提供的雨算法(python版本)。

简介：
去雨算法的Python实现，降雨条纹会对可见度造成严重的损害，从而导致许多现有的计算机视觉算法无法正常运行。因此，从图像中去除降雨变得至关重要。我们提出了一种基于深度卷积和循环神经网络的新型深度网络架构，用于单图像去雨。由于上下文信息对于降雨去除至关重要，因此我们首先采用扩张卷积神经网络来获得较大的感受野。为了更好地适应降雨去除任务，我们还对网络进行了修改。在强降雨条件下，降雨条纹具有各种方向和形状，可以被视为多个降雨条纹层级的累积。通过引入挤压与兴奋模块，我们为不同的降雨条纹层级分配了不同的α值，根据强度和透明度进行分配。由于降雨条纹层级相互重叠，一次性完全去除降雨并非易事。因此，我们进一步将降雨去除分解为多个阶段。循环神经网络被整合进来，以保留先前阶段有用的信息并为后续的降雨去除提供帮助。我们在合成数据集和真实世界数据集上进行了广泛的实验。我们的所提出方法在所有评估指标下均优于最先进的方法。