论文标题

单图像连续降雨密度估计

Single Image Deraining with Continuous Rain Density Estimation

论文作者

He, Jingwei, Yu, Lei, Xia, Gui-Song, Yang, Wen

论文摘要

由于雨水密度不均匀和多种雨滴尺度,单图(SIDR)通常遭受过度/下方的影响。在本文中,我们为sidr提出了一个\ textbf {\ it co} ntinous \ textbf {\ it de} nsity引导网络(代码net)。特别是,它由{{\ color {black}条纹}提取器和一个denoiser}组成,其中利用了卷积稀疏编码(CSC)来过滤从提取的雨条中滤出噪音。受到CSC的重新迭代软阈值的启发,我们通过学习稀疏代码的频道注意力块来解决连续雨密度估计的问题。我们进一步{\ color {black}开发}一种多尺度策略,描绘了以不同尺度出现的雨条。关于合成和现实世界数据的实验证明了我们的方法优于最近的{\ color {black {black {black {black {black {black {black {black {black {black {black {black {black {black {black {black {black {black {black {black {black {black}}状态。此外,我们的代码网不用量化多个级别的雨密度,而是可以提供对雨密度的连续值估计,这在实际应用中更为可取。

Single image deraining (SIDR) often suffers from over/under deraining due to the nonuniformity of rain densities and the variety of raindrop scales. In this paper, we propose a \textbf{\it co}ntinuous \textbf{\it de}nsity guided network (CODE-Net) for SIDR. Particularly, it is composed of { a rain {\color{black}streak} extractor and a denoiser}, where the convolutional sparse coding (CSC) is exploited to filter out noises from the extracted rain streaks. Inspired by the reweighted iterative soft-threshold for CSC, we address the problem of continuous rain density estimation by learning the weights with channel attention blocks from sparse codes. We further {\color{black}develop} a multiscale strategy to depict rain streaks appearing at different scales. Experiments on synthetic and real-world data demonstrate the superiority of our methods over recent {\color{black}state of the arts}, in terms of both quantitative and qualitative results. Additionally, instead of quantizing rain density with several levels, our CODE-Net can provide continuous-valued estimations of rain densities, which is more desirable in real applications.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源