ETDNet: An Efficient Transformer Deraining Model

作者：Qin Qin, Jingke Yan, Qin Wang, Xin Wang, Minyao Li, Yuqing Wang · 发表于：IEEE Access · 年份：2021 · DOI：10.1109/access.2021.3108516 · 被引用次数：38 · 研究领域：Image Enhancement Techniques、Image and Signal Denoising Methods、Advanced Vision and Imaging

Rainy days usually degrade the visual effect of images and videos. At present, most deraining models for single images adopt gradual optimization or elimination to remove rain streaks, but actually with relatively low efficiency in real tasks. An efficient one-stage deraining model, Efficient Transformer Derain Network (ETDNet), is proposed to remove rain streaks in single images efficiently. A new Transformer architecture is designed to provide rich multiple scales and context information, making the model extract features in a coarse-to-fine way. Multiple expansion filters with different expansion rates are embedded to predict the suitable kernel for each pixel of the rainy image in a multi-scale way. A multi-scale Loss Function is introduced to restore features with high-fidelity and detail textures. Experiments on Rain100L, Rain100H, and SPA datasets show that compared with other models, the proposed ETDNet reaches the highest PSNR and SSIM values at the fastest speed.