46
GitHub - taki0112/SPADE-Tensorflow: Simple Tensorflow implementation of "Se...
source link: https://github.com/taki0112/SPADE-Tensorflow
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
README.md
Semantic Image Synthesis with SPADE (GauGAN) - Tensorflow
Simple Tensorflow implementation of "Semantic Image Synthesis with Spatially-Adaptive Normalization" (CVPR 2019 Oral)
Project page | Paper | Pytorch code
Preparation
- Image
- Segmentation map
- Don't worry. I do one-hot encoding of segmentation map automatically (whether color or gray).
Usage
├── dataset
└── YOUR_DATASET_NAME
├── image
├── 000001.jpg
├── 000002.png
└── ...
├── segmap
├── 000001.jpg
├── 000002.png
└── ...
├── segmap_test
├── a.jpg
├── b.png
└── ...
├── segmap_label.txt (Automatically created)
├── guide.jpg (example for guided image translation task)
Train
> python main.py --dataset spade_celebA --img_ch 3 --segmap_ch 3 --phase train
Random test
> python main.py --dataset spade_celebA --segmap_ch 3 --phase random
Guide test
> python main.py --dataset spade_celebA --img_ch 3 --segmap_ch 3 --phase guide --guide_img ./guide_img.png
Our Results
Loss grpah
CityScape
CelebA-HQ (Style Manipulation)
CelebA-HQ (Random Manipulation)
How about the Least-Square loss ?
CelebA-HQ (Style Manipulation)
CelebA-HQ (Random Manipulation)
Architecture
Generator Image Encoder Discriminator All-in-oneSPADE architecture
SPADE SPADE Residual BlockAuthor
Recommend
About Joyk
Aggregate valuable and interesting links.
Joyk means Joy of geeK