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

# how to train

1> 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 'stage2/'

2> python train.py --name stage2 --continue_train --using_model_KD  --load_size 256 --crop_size 256 --dataset_mode custom --label_nc 19 --no_instance --batchSize 1 --gpu_ids 0 --checkpoints_dir checkpoint --results_dir result  --checkpoints_from_kd_dir ../stage1/checkpoints/stage1_test   --label_dir datasets/500_triplet/train/labels --image_dir datasets/500_triplet/train/images 












