【摘要】 针对现有的图像去雨算法通常需要设计先验知识和多阶段优化框架导致算法适应场景单一的问题，提出一种基于图像过滤统一去除雨线和雨滴的去雨核预测网络(RKPN)。有雨图像通过RKPN估计像素级去雨内核。采用空洞卷积提取多尺度特征，提出一个多分支特征聚合模块。通过数据增强构建一个雨线和雨滴混合数据集(RDRS),提升现实世界复杂场景中图像去雨算法的效果。对4个公开数据集和RDRS数据集的广泛实验结果表明，所提模型取得均高于MPRNet等主流网络的峰值信噪比和结构相似度。更多还原

【Abstract】 To solve the problem that existing deraining algorithms usually need to design prior knowledge and multi-stage optimization framework to adapt to a single scene, an image filtering-based deraining network(RKPN) was proposed to uniformly remove rain streaks and raindrops. The image with rain was estimated by RKPN to gain the pixel-level deraining kernel. Dilated convolution was used to extract multi-scale features, and a multi-branch feature aggregation module was proposed. A rain streaks and raindrops dataset(RDRS) was constructed through data enhancement to improve the effectiveness of image derain-removal algorithms in real-world complex scenes. Extensive experimental results on four existing datasets and RDRS dataset show that the proposed model achieves higher SSIM and PSNR than mainstream networks such as MPRNet.更多还原