[发明专利]一种视频图像雨滴去除方法及装置

说明书

技术领域

本发明涉及视频图像除雨技术领域，尤其涉及一种视频图像雨滴去除方法及装置。

背景技术

当前，户外计算机视觉系统已经被广泛应用于军事国防、医疗、智能交通等技术领域。但是，由于恶劣天气等环境原因，可能会严重影响户外计算机视觉系统的性能，甚至导致其完全失效。因此消除恶劣天气影响的有效方法，对于一个全天候的户外计算机视觉系统来说必不可少。在众多恶劣天气情况中，雨滴由于拥有较大粒子半径及其他复杂物理特性，会对视觉系统所摄取的图像的质量造成较大程度的影响。图像雨滴去除技术通过使用雨的物理、频率等特性，对图像中的雨滴进行识别、去除。其不仅能够显著提升图像质量，还有利于图像的进一步处理。因此，图像雨滴去除技术已经成为计算机视觉领域不可缺少的关键性技术。

近些年来关于图像中雨滴检测与去除的研究已然成为热点。当前在图像雨滴去除中，现有技术大多采用值简单替换法，即通过背景像素的灰度值来替换探测到的雨滴像素，从而重新构成图像，进行雨滴去除。然而，由于雨滴会对图像或视频造成模糊作用，简单的值替换法会对输出视频图像的质量大大降低。

发明内容

本发明的实施例提供一种视频图像雨滴去除方法及装置，以解决当前雨滴会对图像或视频造成模糊作用，简单的值替换法会对输出视频图像的质量大大降低的问题。

为达到上述目的，本发明采用如下技术方案：

一种视频图像雨滴去除方法，包括：

获取视频中的各帧初始图像，并将各帧初始图像从RGB色彩空间转换为YCbCr色彩空间，获取色彩空间转换后的初始图像的Y分量；

将所述初始图像的Y分量进行二维经验模态分解，生成所述初始图像的轮廓图像；

将所述初始图像的Y分量中的每个像素点进行双边滤波，生成物体显著边缘图像；

将所述物体显著边缘图像和所述轮廓图像进行二值化并相交处理，生成去雨部分图像；

根据小波变换模极大值算法对所述去雨部分图像进行小波去噪处理，生成去噪后图像；

根据去噪后图像的Y分量和所述初始图像的Y分量进行α混合，生成YCbCr色彩空间下的结果图像；

将所述YCbCr色彩空间下的结果图像从YCbCr色彩空间转换为RGB色彩空间，形成RGB色彩空间下的结果图像；

将各帧RGB色彩空间下的结果图像进行合成，生成去雨视频。

具体的，所述将各帧初始图像从RGB色彩空间转换为YCbCr色彩空间，获取色彩空间转换后的初始图像的Y分量，包括：

通过公式：