文献类型：期刊文献

英文题名：CONTEXT-AWARE TRANSFORMER FOR SINGLE IMAGE RAIN STREAKS REMOVAL

作者：Chen, Zhihua[1]; Liang, Lei[1]; Huang, Yeting[1]; Dai, Lei[1]; Li, Ran[1]; Sheng, Bin[2]

机构：[1] Department of Computer Science and Engineering, East China University of Science and Technology, Shanghai, China; [2] Department of Computer Science and Engineering, Shanghai Jiao Tong University, Shanghai, China

年份：2024

起止页码：7650

外文期刊名：ICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing - Proceedings

收录：EI(收录号：20242416239606)

语种：英文

摘要：Deep learning based image deraining has been widely researched. However, rain streaks are hard to differentiate with similar textures of background without context knowledge. In this paper, a novel Context-Aware Transformer (CAT) is proposed for single image deraining where both local and global context within the input rainy image are utilized for better background reconstruction performance. The proposed CAT perceives a comprehensive context view through efficient self-attention mechanism and dilated convolutions in the Context-Aware Transformer Block (CATB). The Rain-Aware Feature Selection module (RAFS) generates feature blending coefficients adaptively to filter out rain streaks components and preserves clear background in hierarchical features of CAT. Meanwhile, a High-Frequency Preserved Loss (HFPL) provides further supervision on training and promotes reserving clearer structures and sharper details. Experiments on synthesized and real-world benchmarks illustrate the outstanding performance over state-of-the-art methods and pleasing visual results in various scenes. ? 2024 IEEE.