一种基于雨势尺度的视频雨滴去除算法

摘要

本发明旨在解决上述现有技术中存在的问题，提出一种基于雨势尺度的视频雨滴去除算法，包括以下步骤：S100、获取视频像素总数P，确定样本像素点的标号；S200、获取视频前N帧对应像素的亮度平均值；S300、求出前N帧图像中每个像素的最大值及最小值；S400、采用K均值聚类方法，根据每个像素亮度值将像素分为雨滴或背景，标记被雨滴污染的帧数并求和sum；S500、判断sum/(M×N)是否大于预设阈值T，是则采用帧差法处理被雨滴覆盖像素；否则采用引导滤波器处理被雨滴覆盖像素。本发明在视频去雨滴处理之前进行预处理，判断雨势大小，再选择合适的算法去除雨滴，能够获得更好的雨滴去除效果。

主权项

一种基于雨势尺度的视频雨滴去除算法，包括以下步骤：S100、获取视频像素总数P，并确定样本像素点的标号；S200、获取视频前N帧对应像素的亮度平均值；S300、求出所述前N帧图像中每个像素的最大值以及最小值；S400、以所述最大值以及最小值作为聚类中心采用K均值聚类方法，根据每个像素的亮度值将像素分为雨滴或背景两类，标记出被雨滴污染的帧数n1至nM存储于数组n[M]中，并求和sum=n[0]+…+n[M‑1]，其中，M为所述前N帧图像中样本像素的个数，sum表示M个样本像素在N帧视频中被雨覆盖的次数；S500、判断sum/(M×N)是否大于预设阈值T，是则判定为小雨势并执行步骤S601；否则判定为大雨势并执行步骤S602；S601、采用帧差法处理被雨滴覆盖的像素；S602、采用引导滤波器处理被雨滴覆盖的像素。