ETDNet: An Efficient Transformer Deraining Model

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 Transf...

Descripción completa

Guardado en:
Detalles Bibliográficos
Autores principales: Qin Qin, Jingke Yan, Qin Wang, Xin Wang, Minyao Li, Yuqing Wang
Formato: article
Lenguaje:EN
Publicado: IEEE 2021
Materias:
Acceso en línea:https://doaj.org/article/511042f6fc5b44a98ef222d8f3bf542f
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
Descripción
Sumario: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 the proposed ETDNet reaches the highest PSNR and SSIM values at the fastest speed compared with other models.