STGAN: A Unified Selective Transfer Network for Arbitrary Image Attribute Editing

Ming Liu, Yukang Ding, Min Xia, Xiao Liu, Errui Ding +2 more
4/22/2019
cs.CV

Official code: csmliu/STGAN (441 stars, MIT, last commit 2020-04-19).

Abstract

Arbitrary attribute editing generally can be tackled by incorporating encoder-decoder and generative adversarial networks. However, the bottleneck layer in encoder-decoder usually gives rise to blurry and low quality editing result. And adding skip connections improves image quality at the cost of weakened attribute manipulation ability. Moreover, existing methods exploit target attribute vector to guide the flexible translation to desired target domain. In this work, we suggest to address these issues from selective transfer perspective. Considering that specific editing task is certainly only related to the changed attributes instead of all target attributes, our model selectively takes the difference between target and source attribute vectors as input. Furthermore, selective transfer units are incorporated with encoder-decoder to adaptively select and modify encoder feature for enhanced attribute editing. Experiments show that our method (i.e., STGAN) simultaneously improves attribute manipulation accuracy as well as perception quality, and performs favorably against state-of-the-arts in arbitrary facial attribute editing and season translation.

View on arXivView PDF

Code Implementations(1)

csmliu/STGANOfficial100%
MIT

STGAN: A Unified Selective Transfer Network for Arbitrary Image Attribute Editing

44184PythonApr 2, 20196 years agoMIT

Similar Papers

Cite this paper

@article{liu2019stgan,
  title  = {STGAN: A Unified Selective Transfer Network for Arbitrary Image Attribute Editing},
  author = {Ming Liu and Yukang Ding and Min Xia and Xiao Liu and Errui Ding and Wangmeng Zuo and Shilei Wen},
  year   = {2019},
  eprint = {1904.09709},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/1904.09709}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/1904.09709.svg)](https://codeofpaper.com/paper/1904.09709)

Discussion