一种基于注意力生成对抗网络的图像去雨滴方法

摘要

本发明公开了一种基于注意力生成对抗网络的图像去雨滴方法，将待去雨滴的原始带雨滴图像输入到注意力循环网络获得注意力图，所述注意力循环网络包括多个子网络，首个子网络的输入为原始带雨滴图像与初始注意力图的拼接结果，前一个子网络的输出与原始带雨滴图像进行拼接后输入到下一个子网络，每个子网络包括残差层、长短时记忆网络层和卷积神经网络层；然后将原始带雨滴图像和注意力图输入到上下文自动编码器，生成去雨滴之后的图像。本发明技术方案在图像雨滴较少、雨滴造成图像模糊以及图像雨滴密集等情况下，都能达到较好去雨滴效果。