基于SPCANet的单幅图像去雨方法研究
Research on Single Image Deraining Method Based on SPCANet
-
摘要: 针对真实雨纹空间分布随机、提取雨纹特征时背景细节丢失的问题,提出一种基于SPCANet(Spatial and Channel Attention Network)的雨天环境下指针式仪表示数读取算法。首先,在SPANet(Spatial Attentive Network)网络基础上引入通道注意力机制构建增强型Bottleneck模块,提出了SPCANet;随后,采用多尺度特征提取与融合策略结合混合注意力机制,对空间注意力模块进行优化,并设计复合损失函数,从多个维度对模型训练进行约束。最后,通过协同应用上述改进,提升图像去雨模型在复杂场景下的整体性能。实验结果表明,所建方法能有效去除雨纹的同时还能更好地恢复图像背景细节信息。Abstract: Aiming at the problems of the random spatial distribution of real raindrop streaks and the loss of background details during the extraction of raindrop streak features, an algorithm for reading the indications of pointer instruments in rainy environments based on SPCANet (Spatial and Channel Attention Network) is proposed. SPCANet introduces a channel attention mechanism on the basis of the SPANet (Spatial Attentive Network) network to construct an enhanced Bottleneck module. It adopts a multi-scale feature extraction and fusion strategy, combines a hybrid attention mechanism, and optimizes the spatial attention module. A composite loss function is designed to constrain the model training from multiple dimensions. These improvement measures work synergistically, aiming to significantly enhance the overall performance of the image deraining model in complex scenarios. Experimental results show that the proposed method can effectively remove raindrop streaks and better restore the detailed information of the image background.
下载: