首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

Removing Rain from SI via Deep Detail Network

Fu X, Huang J, Huang D Z Y, et al. Removing rain from single images via a deep detail network[J].

当object的structure, orientation与rain streak相似时,基于low-level图像特征的de-rain方法效果很差

ResNet通过改变映射形式简化训练过程

因此,论文提出用于de-rain的Deep Detail Network框架

利用lossless negative residual mapping, 缩小映射范围,简化训练过程.

(输入)rain image - (预测)rain residual = (输出)clean image

引入prior信息(rain high-frequency).

去除图片背景,得到包含object and rain streaks structures的图片,将其作为神经网络的输入,神经网络对rain structure建模.

此外,去除背景后的图片像素值范围缩小,且大部分像素点为0,其稀疏性能够提高效果。

Y: gt; X: rain image; X_detail: obj and rain streak structure; |Y - X|: rain structure.

创建数据集.

框架特点

泛化到真实图像

通用性. 能用于去噪,降低JPEG artifacts

基于先验信息缩小映射范围(而ResNet改变映射形式)

数据集

UCIDBSDGoogle image search中收集1000张clean image

每张图片生成14种不同streak orientation and magnitude的rain image

相关工作

分为两类

基于视频

基于单张图片

基于视频的de-rain任务可以使用inter-frame information,相对于基于单张图片而言,较简单。

基于单张图片的工作有:

Non-local mean fltering

GMM + patch-rank prior(low-rank)

(patch-based)

GMM + patch-based prior

DSC(discriminative sparse coding). Dictionary Learning

Findings

heavy rain image存在类似于haze的现象,因此,对图片进行de-haze预处理能够提高效果。

1. Deep Detail Network

Direct Network

神经网络输入(X): rain image

神经网络输出>: clean image

出现color shift现象.

frobenius范数

平方-求和-开方

Negative Residual Mapping

神经网络输入(X): rain image

神经网络输出: rain structure

模型输出: clean image = rain image - rain structure

gt: clean image

通过缩小输出映射范围,减小solution space. 无法完全de-rain.

Deep Detail Network

神经网络输入(X_detail): obj + rain structure image

神经网络输出: rain structure

模型输出: clean image = rain image - rain structure

gt: clean image

输入具有稀疏性 + 输出缩小映射范围.

神经网络为Parameter Layers

实验中网络深度L设为26

Decompose

使用guided filtering(实验中,半径设为15)作为low-pass filter,将图片分为

Base layer

Detail layer

Detail layer包含图片中object and rain streak structure信息.

网络结构

2.Experiments

实验与两个算法进行对比

[25] DSC

[24] GMM + Patch-rank prior

评价指标SSIM

合成数据

真实数据

测试时间

[24] [25]算法为CPU运行时间。

收敛比较

网络depth vs width

与深度学习算法(ICCV 2013)比较

扩展任务(通用性)

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20190221G1E0KV00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码

添加站长 进交流群

领取专属 10元无门槛券

私享最新 技术干货

扫码加入开发者社群
领券