基于深度学习的图像去雨（Pytorch，Opencv）

最近做了一个图像去雨项目，使用的语言为Python，深度学习框架为Pytorch。去雨效果不错。文章最后附有代码下载地址。

程序运行效果：

效果视频：b站视频地址

网络模型：

去雨的网络架构为Unet模型，模型的结构图如下所示：

数据集：

去雨数据集为rainL，共有1800张有雨图像，每张有雨图像都对应一张无雨图像，如下图所示：

项目结构如下图所示：

运行main.py即可弹出界面。

项目下载：项目地址