[发明专利]一种基于稀疏表示的单幅图像去雨方法

摘要

本发明涉及一种基于稀疏表示的单幅图像去雨方法，包括得到原始图像的低频部分和高频部分；对高频部分进行无重叠的分块；并做在线字典学习，得到所对应的由多个子块构成得字典DHF；利用近邻传播聚类方法将DHF每个子块分为两类，一类表示高频部分的几何分量信息，另一类表示高频部分中雨分量信息；提取每个子块的颜色分布和边缘方向，对已经形成的高频部分的雨字典的每个子块再做一次分类；恢复高频部分中的雨分量；得到高频部分的几何分量；得到去雨后的图像。本发明具有较好去雨效果。

主权项

一种基于稀疏表示的单幅图像去雨方法，包括下列步骤：步骤1：输入原始图像I；步骤2：利用双边滤波器得到原始图像的低频部分ILF和高频部分IHF；步骤3：对高频部分IHF进行无重叠的分块bk∈Rn,k＝1,2,...,p，然后计算每块的HOG特征值；步骤4：对IHF做在线字典学习，得到IHF对应的字典DHF，DHF由多个子块构成，并得到字典学习的稀疏系数；步骤5：利用近邻传播聚类方法将DHF每个子块分为两类，一类表示高频部分的几何分量信息，另一类表示高频部分中雨分量信息，形成高频部分的雨字典其中，几何分量具有较复杂的信息，从而子块的几何分量方差应该比雨分量方差大，方差大的一类为几何分量；步骤6：提取每个子块的颜色分布和边缘方向，对已经形成的高频部分的雨字典的每个子块再做一次分类，使经过步骤5中误分类或遗漏的子块能够被正确地分类；步骤7：根据步骤4中字典学习的稀疏系数，恢复高频部分IHF中的雨分量步骤8：用并进行双边滤波和BM3D平滑，最终的结果作为高频部分的几何分量步骤9：通过公式得到去雨后的输出图像。