Single-Image Snow Removal Based on an Attention Mechanism and a Generative Adversarial Network

Aiwen Jia, Zhenhong Jia, Jie Yang, Nikola Kasabov

Research output: Contribution to journalArticlepeer-review

5 Citations (Scopus)
114 Downloads (Pure)


Bad weather, such as snowfall, can seriously decrease the quality of images and pose great challenges to computer vision algorithms. In view of the negative effect of snowfall, this paper presents a single-image snow removal method based on a generative adversarial network (GAN). Unlike previous GANs, our GAN includes an attention mechanism in the generator component. By injecting attention information, the network can pay increased attention to areas covered by snow and improve its capability to perform local repairs. At the same time, we improve the traditional U-Net network by combining it with the residual network to enhance the effect of the model when removing snowflakes from a single image. Our experiments on both synthetic and real-word images show that our method produces better results than those of other state-of-the-art methods.
Original languageEnglish
Pages (from-to)12852-12860
Number of pages9
JournalIEEE Access
Early online date13 Jan 2021
Publication statusPublished (in print/issue) - 22 Jan 2021


  • snow removal
  • generative adversarial networks
  • attention mechanisms


Dive into the research topics of 'Single-Image Snow Removal Based on an Attention Mechanism and a Generative Adversarial Network'. Together they form a unique fingerprint.

Cite this