This code is based on SEAN(https://github.com/ZPdesu/SEAN) and style-based-gan-pytorch(https://github.com/rosinality/style-based-gan-pytorch). Please install the dependencies from both projects. 

Note in SEAN, the project is SPADE if style-branch is ignored
# Note

# train
1> Downloading the pretrained StyleGAN (i.e., AFHQ.model) on AFHQ dataset for the link(https://drive.google.com/drive/folders/1JdDJz3RX6DVl_dkK5SgNj_iLZt9DmE64?usp=sharing). Moving the pretrained model into 'stylegan_model'

2> Downloading the dataset (i.e., datasets.zip) form the link(https://drive.google.com/drive/folders/1I-kyMSDENnpUrcJNUpwFRk98TsQh1ph1?usp=sharing). uncompressing and moving it into 'stage1/'


3> python train.py --name stage1 --load_size 256 --crop_size 256 --dataset_mode custom --label_dir datasets/500_triplet/train/labels --image_dir datasets/500_triplet/train/images --label_nc 19 --no_instance --batchSize 4 --gpu_ids 0
