Home

Awesome

DRIT-Tensorflow

Simple Tensorflow implementation of Diverse Image-to-Image Translation via Disentangled Representations (ECCV 2018 Oral)

<img src='./assets/final.gif' width="800px">

Pytorch version

Requirements

Usage

Download Dataset

├── dataset
   └── YOUR_DATASET_NAME
       ├── trainA
           ├── xxx.jpg (name, format doesn't matter)
           ├── yyy.png
           └── ...
       ├── trainB
           ├── zzz.jpg
           ├── www.png
           └── ...
       ├── testA
           ├── aaa.jpg 
           ├── bbb.png
           └── ...
       └── testB
           ├── ccc.jpg 
           ├── ddd.png
           └── ...
           
├── guide.jpg (example for guided image translation task)

Train

python main.py --phase train --dataset summer2winter --concat True

Test

python main.py --phase test --dataset summer2winter --concat True --num_attribute 3

Guide

python main.py --phase guide --dataset summer2winter --concat True --direction a2b --guide_img ./guide.jpg

Tips

Summary

Comparison

comparison

Architecture

true false

Train phase

train_1 train_2

Test & Guide phase

test

Results

result_1 result_2

Related works

Author

Junho Kim