scripts/train/train_8x_128x128.sh
#!/bin/bash
PYTHON_ENV=YOUR_PYTHON_PATH/bin

# This should contain the path to the cloned DeepSEE repository
DEEPSEE_PATH=YOUR_DEEPSEE_PATH
cd $DEEPSEE_PATH

IMG_DIR_TRAIN=YOUR_TRAIN_IMG_PATH
LABEL_DIR_TRAIN=YOUR_TRAIN_LABEL_PATH
IMG_DIR_VAL=YOUR_VAL_IMG_PATH
LABEL_DIR_VAL=YOUR_VAL_LABEL_PATH

# This is a file with identity annotations for each image. You can download it
# from the CelebA website ("CelebA" -> "Anno" -> "identity_CelebA.txt")
# You only need this for the guided model, so you can leave this empty for the independent model.
IDENTITIES_FILE=YOUR_PATH_TO_IDENTITIES_FILE/identity_CelebA.txt
DATASET=celeba
# Give your GPU id(s) here or use -1 if none.
GPU_IDS=0

# Choose the model you want to train by uncommenting
NAME=CelebA_8x_independent_128x128
#NAME=CelebA_8x_guided_128x128

$PYTHON_ENV/python $DEEPSEE_PATH/train.py \
    --name $NAME \
    --dataset_mode $DATASET \
    --image_dir $IMG_DIR_TRAIN --label_dir $LABEL_DIR_TRAIN \
    --image_dir_val $IMG_DIR_VAL --label_dir_val $LABEL_DIR_VAL \
     --evaluate_val_set \
     --tf_log \
    --niter 5 --niter_decay 3 \
    --identities_file $IDENTITIES_FILE \
    --gpu_ids $GPU_IDS \
    --load_config_from_name