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
Summary
Results (128x128)
Women
Men
Related works
Reference
Author
HOANG Duc Thang