一种基于卷积神经网络的无人机单幅图像去雨方法

摘要

本发明公开了一种基于卷积神经网络的无人机单幅图像去雨方法，首先，通过引导滤波器将有雨图像分离成低频背景图像和高频雨线图像；然后，利用一个多尺度的CNN卷积神经网络对低频背景图像和高频雨线图像进行分通道的特征提取；接着，通过通道域注意力机制对多尺度的CNN卷积神经网络提取出的特征图重新分配权重；然后，通过一个全连接层对多个通道的特征图进行拼接，构建出有雨图像和去雨图像之间的映射关系；最后，分别输出高、低频的去雨图像，将二者融合输出完整的去雨图像。本发明具有高效去除无人机飞行过程中拍摄到雨天图像上雨线的特点。