[发明专利]基于双通道多尺度判别模型去雨系统及控制方法

说明书

本发明公开了一种基于双通道多尺度判别模型去雨系统及控制方法，所述系统包括：生成器和双通道多尺度判别器，所述生成器用于生成对抗网络，所述生成器具有多尺度特征提取单元和密集连接单元组成，所述多尺度特征提取单元用于获取接收场，并根据尺度的输入获取多尺度雨痕信息；所述密集连接单元用于去除雨层保留背景层；所述双通道多尺度判别器具有双通道判别模块，所述双通道判别模块用于判断输出去雨图像和与真实无雨图像真伪。其利用双重判别模型生成对抗网络解决雨滴对图像的影响，增强去雨效果，解决现有技术中相机在雨天拍摄的图像严重退化，严重影响图像的视觉质量的问题。

技术领域

本发明涉及计算机视觉领域，具体涉及一种基于双通道多尺度判别模型去雨系统及控制方法。

背景技术

随着科学技术的快速发展和智能手机的普及，手机和摄像头在雨天拍摄的图像严重退化，严重影响图像的视觉质量。为了提高这些退化图像的整体质量，以获得更好的视觉效果，并确保视觉算法的性能得到增强，自动去除这些不希望出现的雨滴变得至关重要，因此图像去雨是一个十分重要的课题。

目前现有的去雨的方法分成两类：一种是基于传统方法的去雨，例如基于高频的方法，假设雨条属于输入图像的高频分量，然后通过稀疏编码和字典学习进一步将其分为雨层和非雨层。另一种是基于卷积神经网络方法去雨，利用 CNN结构直接学习输入的雨景图像与其对应的地面真实性之间的非线性映射，并在一定的先验信息下，实现了良好的视觉效果，例如多流密集连接网络。

目前基于生成对抗网络的图像去雨方法中，通过密集连接生成器和多尺度判别模型进行去雨，但是现实世界雨密度的不均匀性且雨的尺度也不一样，单图像雨条纹的去除是一个极具挑战性的问题；此外，为了确保生成图像与真实世界图像保持一致，单一判别模型存在局限性。

发明内容

为了解决现有技术中相机在雨天拍摄的图像严重退化，严重影响图像的视觉质量的问题，本发明实施例提供了一种基于双通道多尺度判别模型去雨系统及控制方法。所述技术方案如下：

第一方面，提供了一种基于双通道多尺度判别模型去雨系统，所述系统包括：生成器和双通道多尺度判别器，其中：

所述生成器用于生成对抗网络，所述生成器具有多尺度特征提取单元和密集连接单元组成，所述多尺度特征提取单元用于获取接收场，并根据尺度的输入获取多尺度雨痕信息；所述密集连接单元用于去除雨层保留背景层；

所述双通道多尺度判别器具有双通道判别模块，所述双通道判别模块用于判断输出去雨图像和与真实无雨图像真伪。

进一步的，所述密集连接单元中的密集块之间跳跃连接。

进一步的，每个所述密集块后连接有过渡块。

进一步的，所述双通道多尺度判别器还具有多尺度池化模块，所述多尺度池化模块与双通道判别模块连接，所述多尺度池化模块用于对特征进行多尺度池化。

第二方面，提供了一种基于双通道多尺度判别模型去雨的控制方法，所述方法包括：

将数据集预处理后，输入至生成器；

所述生成器中的多尺度特征提取单元获取数据集，提取多尺度特征并传入相互连接的所述多尺度特征连接后馈送至所述生成器中的密集连接单元；

所述密集连接单元去除雨层后将去雨图像输送至双通道多尺度判别器中与双通道判别模块连接的多尺度池化模块进行池化；

将真实无雨图像输送至所述双通道多尺度判别器；所述双通道多尺度判别器中的所述双通道判别模块将去雨图像和真实无雨图像进行比对，判断去雨图像与真实无雨图像真伪，得到真伪结果；

将所述真伪结果输出。