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

Multimodal Unsupervised Image-to-Image Translation

About

Unsupervised image-to-image translation is an important and challenging problem in computer vision. Given an image in the source domain, the goal is to learn the conditional distribution of corresponding images in the target domain, without seeing any pairs of corresponding images. While this conditional distribution is inherently multimodal, existing approaches make an overly simplified assumption, modeling it as a deterministic one-to-one mapping. As a result, they fail to generate diverse outputs from a given source domain image. To address this limitation, we propose a Multimodal Unsupervised Image-to-image Translation (MUNIT) framework. We assume that the image representation can be decomposed into a content code that is domain-invariant, and a style code that captures domain-specific properties. To translate an image to another domain, we recombine its content code with a random style code sampled from the style space of the target domain. We analyze the proposed framework and establish several theoretical results. Extensive experiments with comparisons to the state-of-the-art approaches further demonstrates the advantage of the proposed framework. Moreover, our framework allows users to control the style of translation outputs by providing an example style image. Code and pretrained models are available at https://github.com/nvlabs/MUNIT

Xun Huang, Ming-Yu Liu, Serge Belongie, Jan Kautz• 2018

Related benchmarks

TaskDatasetResultRank
Semantic segmentationFoggy Driving (FD) (test)
mIoU47.8
56
Semantic Image SynthesisCityscapes
FID84
54
Semantic segmentationFoggy Zurich (test)
mIoU39.1
51
Across-modality synthesis (T2-weighted MRI to CT)Pelvic MRI-CT dataset (test)
PSNR23.44
42
Image-to-Image TranslationCelebA-HQ
FID56.8
28
Photo RetouchingMIT Adobe FiveK
PSNR20.32
25
Image-to-Image TranslationEdges2Shoes (test)
FID56.2
24
Photorealistic Image-to-Image TranslationMIT-Adobe FiveK (test)
Inference Latency (s)0.336
24
Image-to-Image TranslationHorse -> Zebra
FID133.8
23
Pedestrian DetectionLLVIP (test)
mAP@5013.6
20
Showing 10 of 103 rows
...

Other info

Code

Follow for update