最近做了一个图像去雨项目,使用的语言为Python,深度学习框架为Pytorch。去雨效果不错。文章最后附有代码下载地址。
程序运行效果:
效果视频:b站视频地址
网络模型:
去雨的网络架构为Unet模型,模型的结构图如下所示:
数据集:
去雨数据集为rainL,共有1800张有雨图像,每张有雨图像都对应一张无雨图像,如下图所示:
项目结构如下图所示:
运行main.py即可弹出界面。
项目下载:项目地址
最近做了一个图像去雨项目,使用的语言为Python,深度学习框架为Pytorch。去雨效果不错。文章最后附有代码下载地址。
程序运行效果:
效果视频:b站视频地址
网络模型:
去雨的网络架构为Unet模型,模型的结构图如下所示:
数据集:
去雨数据集为rainL,共有1800张有雨图像,每张有雨图像都对应一张无雨图像,如下图所示:
项目结构如下图所示:
运行main.py即可弹出界面。
项目下载:项目地址