一种基于雨线主方向自适应的全局稀疏去雨模型

作者：王科平; 蔡凯利^*; 王红旗; 杨艺

来源：智能系统学报, 2020, 15(02): 271-280.

• 单幅图像去雨
• 雨线主方向
• 图像分块
• HOG特征
• 全局稀疏模型
• 稀疏正则项
• 颜色掩膜
• 通道重组

针对现有单幅图像去雨算法较少考虑风力对雨线主方向产生影响的问题,当雨线偏离垂直方向时,现有方法未采取旋转或者只进行大致旋转,去雨后的结果图出现了雨线残留明显或背景模糊化的现象。因此,本文提出了一种基于雨线主方向自适应的全局稀疏去雨模型。首先,将方差最小的图像块与图像库的雨线图依据HOG特征进行匹配,将匹配度最高的雨线图主方向视作待处理雨图的雨线主方向,从而确定全局稀疏模型的旋转角。然后,通过3个包含旋转角的稀疏正则项构建全局稀疏模型进行去雨。最后,通过颜色掩膜对全局稀疏模型去雨后的部分背景进行保护,再与原来的CbCr通道重组,得到最终的去雨图。研究结果表明:在峰值信噪比和结构相似性方面,本文算法均比3种典型的对比算法有所提高,且运行时间相对短。本文算法在有效去除雨线的同时,可较大程度地保留图像的背景细节信息。

• 单位
  自动化学院; 河南理工大学