1.pytorch 实现的图像去雨代码,附详细注释和实验操作流程,代码可以直接运行
2.利用 Rain100 已经有训练和测试数据集,也可以参考 readme 文档更换自己的数据集
3.可以直接运行查看去雨后的效果,代码包含在 Rain100H,Rain100L,Rain1400 上已经训练好的网络参数,需要测试哪个数据集只需要更换测试图片和参数文件即可
4.可以训练自定义的网络结构或者数据集
5.readme 文档中有详细的环境和包
6.附有计算 psnr 和 ssim 的代码
7.一次下载基本包含图像去雨涉及到的各方面,欢迎下载。