基于循环内外密集连接和注意力增强的单幅图像去雨方法

作者:傅予力; 霍万良; 蔡磊; 向友君; 夏君君; 张靓; 王鹏程
来源:2023-02-17, 中国, CN202310127850.7.

摘要

本发明公开了一种基于循环内外密集连接和注意力增强的单幅图像去雨方法,包括:构建用于单幅图像去雨的循环内外密集连接和注意力增强网络,该网络包括多次共享参数的循环子网络,确定损失函数,损失函数包括结构相似度和感知损失;输入训练数据,训练网络参数得到循环去雨模型。本发明在循环内建立密集连接,拓展网络信息传递的通道,使包含更多背景细节信息的浅层特征通过信息传递被网络深层和后续循环的卷积层充分利用,让网络循环渐进式去雨过程中能够保留更多的背景细节。