一种视频图像雨滴去除方法及装置

摘要

本发明提供了一种视频图像雨滴去除方法及装置，涉及视频图像除雨技术领域。方法包括：获取视频中的各帧初始图像，从RGB色彩空间转换为YCbCr色彩空间，获取色彩空间转换后的初始图像的Y分量；并进行二维经验模态分解，生成初始图像的轮廓图像；进行双边滤波，生成物体显著边缘图像；将物体显著边缘图像和轮廓图像进行二值化并相交处理，生成去雨部分图像；对去雨部分图像进行小波去噪处理，生成去噪后图像；根据去噪后图像的Y分量和初始图像的Y分量进行α混合，生成YCbCr色彩空间下的结果图像，并从YCbCr色彩空间转换为RGB色彩空间，形成RGB色彩空间下的结果图像，从而生成去雨视频。本发明解决了当前简单的值替换法会对输出视频图像的质量大大降低的问题。

主权项

一种视频图像雨滴去除方法，其特征在于，包括：获取视频中的各帧初始图像，并将各帧初始图像从RGB色彩空间转换为YCbCr色彩空间，获取色彩空间转换后的初始图像的Y分量；将所述初始图像的Y分量进行二维经验模态分解，生成所述初始图像的轮廓图像；将所述初始图像的Y分量中的每个像素点进行双边滤波，生成物体显著边缘图像；将所述物体显著边缘图像和所述轮廓图像进行二值化并相交处理，生成去雨部分图像；根据小波变换模极大值算法对所述去雨部分图像进行小波去噪处理，生成去噪后图像；根据去噪后图像的Y分量和所述初始图像的Y分量进行α混合，生成YCbCr色彩空间下的结果图像；将所述YCbCr色彩空间下的结果图像从YCbCr色彩空间转换为RGB色彩空间，形成RGB色彩空间下的结果图像；将各帧RGB色彩空间下的结果图像进行合成，生成去雨视频。