科研之友
微信
新浪微博
Facebook
分享链接
摘要
单幅图像由于没有视频时间序列的关联,并且图像中雨的方向、大小和自然环境的背景的复杂性,使得单幅图像去雨的难度大大增加。为了解决这个问题,提出基于生成对抗网络改进的去雨算法。在训练过程中,由于图像的输入维数比较高,会导致训练模型参数过多等问题,所以采用了Inception模块,该模块降低了输入维度,减少了参数,能够很好地保留图像的细节。在生成网络和判别网络中引入了注意力机制,使生成网络能够很好地关注有雨的部分并恢复除去雨的部分,判别网络能够更好地评判真实图像和生成图像的一致性。
-
单位野战工程学院; 中国人民解放军陆军工程大学
