计算机科学
人工智能
特征(语言学)
特征提取
模式识别(心理学)
判别式
能见度
块(置换群论)
失真(音乐)
编码(集合论)
计算机视觉
图像(数学)
图像融合
数学
地理
带宽(计算)
电信
放大器
哲学
语言学
几何学
集合(抽象数据类型)
气象学
程序设计语言
作者
Wenyin Tao,Xuefeng Yan,Yongzhen Wang,Mingqiang Wei
标识
DOI:10.1109/tim.2023.3346498
摘要
Images captured by vision-based measurement tools often suffer from detail blurring, color distortion, and visibility degradation due to rain streaks in rainy weather. As a potential remedy, we develop a hybrid image deraining network called mixed feature fusion network for single-image deraining (MFFDNet), which artfully integrates local and global image features for better rain removal. The proposed MFFDNet makes three significant contributions. First, MFFDNet takes full advantage of convolutional neural network (CNN) and transformer to produce more discriminative features, improving the model's deraining capacity. Second, we propose a novel local feature extraction module called the channel-spatial attention block (CSAB), which can reduce the interference of rain streaks by separating high-frequency and low-frequency information. Also, this module leverages the spatial attention mechanism to obtain better local features. Finally, we develop a prospective feature fusion module, which can produce local features with global characteristics and global features with local characteristics by fusing these two features. Comprehensive evaluations on four synthetic and two real-world datasets demonstrate that MFFDNet performs well in peak signal-to-noise ratio (PSNR), structural similarity (SSIM), natural image quality evaluator (NIQE), and blind/referenceless image spatial quality evaluator (BRISQUE) metrics. The source code is available at https://github.com/taowenyin/MFFDNet .
科研通智能强力驱动
Strongly Powered by AbleSci AI