StarGAN-Keras

StarGAN-Keras copied to clipboard

Usage

Downloading the dataset

$ bash download.sh celeba

Train

• python main.py --mode train --batch_size <4,8,16... depends on GPU memory>

Test on celebA images

• python main.py --mode test

Test on custom images (please put your images into './samples' directory)

• python main.py --mode custom --custom_image_name <000001.jpg... name of your image> --custom_image_label <1 0 0 1 1... 5 original labels of your image>

Pretrained model

• Download G_weights.hdf5 and put into './models' directory.

Summary

Results (128x128)

Women

Men

Related works

• Keras-GAN
• StarGAN-Tensorflow

Reference

• StarGAN paper
• Author pytorch code

Author

HOANG Duc Thang