Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Diverse Image-to-Image Translation via Disentangled Representations

About

Image-to-image translation aims to learn the mapping between two visual domains. There are two main challenges for many applications: 1) the lack of aligned training pairs and 2) multiple possible outputs from a single input image. In this work, we present an approach based on disentangled representation for producing diverse outputs without paired training images. To achieve diversity, we propose to embed images onto two spaces: a domain-invariant content space capturing shared information across domains and a domain-specific attribute space. Our model takes the encoded content features extracted from a given input and the attribute vectors sampled from the attribute space to produce diverse outputs at test time. To handle unpaired training data, we introduce a novel cross-cycle consistency loss based on disentangled representations. Qualitative results show that our model can generate diverse and realistic images on a wide range of tasks without paired training data. For quantitative comparisons, we measure realism with user study and diversity with a perceptual distance metric. We apply the proposed model to domain adaptation and show competitive performance when compared to the state-of-the-art on the MNIST-M and the LineMod datasets.

Hsin-Ying Lee, Hung-Yu Tseng, Jia-Bin Huang, Maneesh Kumar Singh, Ming-Hsuan Yang• 2018

Related benchmarks

TaskDatasetResultRank
Semantic Image SynthesisADE20K
FID132.2
66
Semantic Image SynthesisCityscapes
FID164
54
Semantic Image SynthesisCOCO Stuff
FID135.5
40
Object DetectionBDD100K (Nighttime)
AP12.68
26
Unpaired Image-to-Image TranslationCat → Dog v1 (test)
FID123.4
14
Image-to-Image TranslationCityscapes
FID155.3
14
Reference-guided image synthesisAFHQ (test)
FID114.8
13
Reference-guided image synthesisCelebA-HQ (test)
FID53.3
12
Image-to-Image Translationselfie2anime
KID15.08
11
Image-to-Image Translationanime2selfie
KID14.85
10
Showing 10 of 38 rows

Other info

Code

Follow for update