HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Diverse Image-to-Image Translation via Disentangled Representations

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

Diverse Image-to-Image Translation via Disentangled Representations

Abstract

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.

Code Repositories

hytseng0509/DRIT_hr
pytorch
Mentioned in GitHub
HsinYingLee/DRIT
Official
pytorch
Mentioned in GitHub
guy-oren/DIRT-OST
pytorch
Mentioned in GitHub
Wenchao-Du/LIR-for-Unsupervised-IR
pytorch
Mentioned in GitHub
HsinYingLee/MDMM
pytorch
Mentioned in GitHub
taki0112/DRIT-Tensorflow
tf
Mentioned in GitHub
Bingwen-Hu/DRIT
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
multimodal-unsupervised-image-to-image-4DRIT
FID: 52.1
multimodal-unsupervised-image-to-image-5DRIT
FID: 95.6
synthetic-to-real-translation-on-gtav-toDomain adaptation
mIoU: 43.2

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Diverse Image-to-Image Translation via Disentangled Representations | Papers | HyperAI