一种基于残差对抗细化网络的两阶段图像去雨方法及系统

作者:邱崧; 黄坤耀; 李庆利; 徐伟; 吴思源; 魏茂麟; 孙力; 胡孟晗; 周梅; 刘洪英
来源:2019-12-18, 中国, ZL201911312059.3.

摘要

本发明公开了一种基于残差对抗细化网络的图像去雨方法,可以有效地去除图像中的雨滴,并且恢复出真实的场景图像。本发明将去雨分为两大阶段,第一阶段主要在于从原始图像中去除残差图像,恢复出较为清晰的图像,再将恢复出的图像输入到第二阶段,恢复被雨滴遮挡部分的背景信息,精修图像,提高质量。本发明将有雨图像看成是清晰图像和残差图像的合成,基于这一原理,通过残差网络从原始有雨图像中得到残差图像,并将原始图像减去残差图像,得到第一阶段的输出图像;在第二阶段中,采用生成式对抗网络将第一阶段的输出图像和残差图像作为输入,其中残差图像作为注意力机制,辅助网络恢复出更真实的图像。本发明还提出了一种图像去雨系统。