一种基于自编码卷积神经网络的单幅图像去雨方法及系统

摘要

本发明涉及一种基于自编码卷积神经网络的单幅图像去雨方法及系统。该方法包括：训练基于自编码卷积神经网络的去雨模型；基于自编码卷积神经网络的去雨模型包括输入层、隐含层以及输出层；隐含层包括4层卷积层以及4层反卷积层；改进卷积层与卷积层之间、反卷积层与反卷积层之间的激活函数；利用跳转连接，在卷积层与反卷积层之间添加权重系数；将待去雨的有雨图像分解为低频基础层以及高频细节层；利用基于自编码卷积神经网络的去雨模型对高频细节层进行去雨处理，生成去雨后的高频细节层；增强去雨后的高频细节层，生成增强后的高频细节层；根据低频基础层与增强后的高频细节层确定去雨后的图像。本发明能够提高去雨效果、解决梯度消失的问题。