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
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | Foggy Driving (FD) (test) | mIoU47.8 | 56 | |
| Semantic Image Synthesis | Cityscapes | FID84 | 54 | |
| Semantic segmentation | Foggy Zurich (test) | mIoU39.1 | 51 | |
| Across-modality synthesis (T2-weighted MRI to CT) | Pelvic MRI-CT dataset (test) | PSNR23.44 | 42 | |
| Image-to-Image Translation | CelebA-HQ | FID56.8 | 28 | |
| Photo Retouching | MIT Adobe FiveK | PSNR20.32 | 25 | |
| Image-to-Image Translation | Edges2Shoes (test) | FID56.2 | 24 | |
| Photorealistic Image-to-Image Translation | MIT-Adobe FiveK (test) | Inference Latency (s)0.336 | 24 | |
| Image-to-Image Translation | Horse -> Zebra | FID133.8 | 23 | |
| Pedestrian Detection | LLVIP (test) | mAP@5013.6 | 20 |