一种基于自监督约束和不成对数据的无监督循环去雨网络方法

    公开(公告)号:CN113554568B

    公开(公告)日:2024-10-18

    申请号:CN202110887952.X

    申请日:2021-08-03

    申请人: 东南大学

    发明人: 王堃宇 黄永明

    摘要: 本发明公开了一种基于自监督约束和不成对数据的无监督循环去雨网络方法。本发明的目标是在给定随机不成对数据的情况下,学习从雨水图片域R到目标干净图片域C的映射。本发明引入了生成对抗思想,主要包括雨水移除子网络、雨水生成子网络、鉴别网络三部分。本发明并没有像其他监督去雨方法一样采用成对的人工合成雨图和背景图进行监督训练,因此可以在一定程度上提高模型处理真实世界雨图时的泛化性,并且也绕开了难以从真实下雨场景中获得用于监督训练的输入‑标签图片对的困境,本发明同时也把加雨引入网络,实现雨纹闭环,为自监督去雨网络提供更强的监督。通过本发明可以实现对雨水的自动去雨,提高图片质量,方便图像进一步处理和利用。