[发明专利]一种融合注意力单幅图像去雨方法

摘要

本发明涉及一种融合注意力单幅图像去雨方法，包括：首先将输入图像进行批归一化处理，然后输入到三维注意力机制与残差密集块结构相结合的编码器网络得到高维特征；使用Transformer机制计算全局特征关联性；使用三维注意力残差密集块构建的解码器将特征矩阵逐级还原到原始输入图像大小，得到输出图像；针对去雨过程中算法导致的图像结构信息和细节高频信息被抹除问题，提出将多尺度结构性损失和常用去雨损失函数结合参与去雨网络训练。本发明针对单幅有雨图像进行去雨都能得到更高质量的去雨图像。本发明结合了三维注意力机制、Transformer和编‑解码器架构，可更好提高网络去雨性能。