4 个月前

清除天空:用于单幅图像去雨的深度网络架构

清除天空:用于单幅图像去雨的深度网络架构

摘要

我们介绍了一种称为DerainNet的深度网络架构,用于从图像中去除雨痕。基于深度卷积神经网络(CNN),我们直接从数据中学习雨天图像和干净图像细节层之间的映射关系。由于我们没有真实世界雨天图像对应的地面真值,因此我们在训练过程中合成了带有雨痕的图像。与其他常见的通过增加网络深度或宽度来提高性能的策略不同,我们利用图像处理领域的知识对目标函数进行修改,从而在使用适度规模的CNN的情况下改善去雨效果。具体而言,我们在细节层(高通)而不是在图像域上训练我们的DerainNet。尽管DerainNet是在合成数据上训练的,但我们发现该网络在测试时对真实世界的图像非常有效。此外,我们还在CNN框架中加入了图像增强技术以进一步提升视觉效果。与当前最先进的单幅图像去雨方法相比,我们的方法不仅提高了雨痕去除的效果,而且在网络训练完成后计算时间显著减少。

基准测试

基准方法指标
single-image-deraining-on-rain100hDerainNet
SSIM: 0.592
single-image-deraining-on-rain100lDerainNet
SSIM: 0.884
single-image-deraining-on-test100DerainNet
SSIM: 0.810
single-image-deraining-on-test1200DerainNet
SSIM: 0.835
single-image-deraining-on-test2800DerainNet
PSNR: 24.31
SSIM: 0.861

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
清除天空:用于单幅图像去雨的深度网络架构 | 论文 | HyperAI超神经