项目采用Swin UNet作为生成器,并结合判别器进行训练,以实现高质量的图像去雨效果。代码注释丰富,确保没有bug且可以运行,适合用于学习和研究。 项目特点 生成模型:使用Swin UNet作为生成器,该模型结合了Transformer的强大特征提取能力和UNet的编码-解码结构,能够有效地去除图像中的雨滴。 数据集:项目支持多种图像去雨数据...