-
U-Net: This is the baseline model without any modifications (see UNet_Rain100L.ipynb).
-
U-Net + Local Attention: This model incorporates local attention in the U-Net architecture (see Attention_Rain100L.ipynb).
-
U-Net + Residual Connections: This model uses residual connections within the U-Net architecture (see Residual_Rain100.ipynb).
-
U-Net + Local Attention + Residual Connections: This model combines both local attention and residual connections (see Attention_Residual_Rain100L.ipynb).
- The "Results - Derained 200 epochs" folder contains all derained test image results produced by each model.



