arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1908.06616cs.CV

SPA-GAN: Spatial Attention GAN for Image-to-Image Translation

Wayne State University(韦恩州立大学)

Hajar Emami, Majid Moradi Aliabadi, Ming Dong, Ratna Babu Chinnam

英文摘要:

Image-to-image translation is to learn a mapping between images from a source domain and images from a target domain. In this paper, we introduce the attention mechanism directly to the generative adversarial network (GAN) architecture and propose a novel spatial attention GAN model (SPA-GAN) for image-to-image translation tasks. SPA-GAN computes the attention in its discriminator and use it to help the generator focus more on the most discriminative regions between the source and target domains, leading to more realistic output images. We also find it helpful to introduce an additional feature map loss in SPA-GAN training to preserve domain specific features during translation. Compared with existing attention-guided GAN models, SPA-GAN is a lightweight model that does not need additional attention networks or supervision. Qualitative and quantitative comparison against state-of-the-art methods on benchmark datasets demonstrates the superior performance of SPA-GAN.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑