arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1802.07412cs.CV
收藏

基于多流稠密网络的密度感知单幅图像去噪

Density-aware Single Image De-raining using a Multi-stream Dense Network

Rutgers University(罗格斯大学)

He Zhang, Vishal M. Patel

摘要:

由于图像中存在不均匀的雨密度,单图像雨条纹去除是一个极具挑战性的问题。提出了一种基于密度感知的多流密集连接卷积神经网络的联合雨密度估计和去雨算法DID-MDN。所提出的方法使网络本身能够自动确定雨密度信息,然后有效地去除相应的雨条纹的估计雨密度标签的指导下。为了更好地表征不同尺度和形状的雨纹,提出了一种多流密集连接的去雨网络,该网络有效地利用了不同尺度的特征。此外,一个新的数据集包含图像与降雨密度标签创建和训练建议的密度感知网络。在合成数据集和真实数据集上的实验表明,该方法比现有的方法有了显著的改进。此外,烧蚀研究,以证明所提出的方法中的不同模块所获得的改进。代码可以在:https://github.com/hezhangsprinter上找到

英文摘要:

Single image rain streak removal is an extremely challenging problem due to the presence of non-uniform rain densities in images. We present a novel density-aware multi-stream densely connected convolutional neural network-based algorithm, called DID-MDN, for joint rain density estimation and de-raining. The proposed method enables the network itself to automatically determine the rain-density information and then efficiently remove the corresponding rain-streaks guided by the estimated rain-density label. To better characterize rain-streaks with different scales and shapes, a multi-stream densely connected de-raining network is proposed which efficiently leverages features from different scales. Furthermore, a new dataset containing images with rain-density labels is created and used to train the proposed density-aware network. Extensive experiments on synthetic and real datasets demonstrate that the proposed method achieves significant improvements over the recent state-of-the-art methods. In addition, an ablation study is performed to demonstrate the improvements obtained by different modules in the proposed method. Code can be found at: https://github.com/hezhangsprinter

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑