taki0112/GDWCT-Tensorflow

Simple Tensorflow implementation of "Image-to-Image Translation via Group-wise Deep Whitening-and-Coloring Transformation" (CVPR 2019 Oral)

Paper | Pytorch code | Slide

Usage

├── 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 --dataset male2female

Test

 > python main.py --dataset male2female --phase test --direction a2b --guide_img ./guide.png

Summary

Coloring

---

Whitening

---

GDWCT

---

Model

---

Process

Results

Loss & Alpha graph

Author

Junho Kim