DeRaindrop

收录理由

雨滴附着在车窗或镜头表面，会严重遮挡背景场景，导致拍摄画面质量大幅下降。这个项目给出了一种基于生成对抗网络的解决方案，能够从单张图像中直接去除雨滴，恢复出干净的图像。其核心思路是在生成器和判别器中都引入视觉注意力机制：生成器借助注意力信息更专注于雨滴区域及其周边结构，判别器则能对修复区域的局部一致性进行有效评估。这种设计不仅解决了雨滴位置未知、遮挡区域信息缺失的难题，也为低层视觉任务中融合注意力机制提供了有价值的参考。对于户外监控、自动驾驶感知等经常受天气干扰的应用场景，该工作具有直接的借鉴意义。代码基于Python 2.7和PyTorch实现，需要CUDA环境，更适合作为研究基线来阅读和改造，而非直接用于生产环境。