Add scripts to download pretrained models

download.sh

@@ -1,14 +1,43 @@
-# CelebA images
-URL=https://www.dropbox.com/s/3e5cmqgplchz85o/CelebA_nocrop.zip?dl=0
-ZIP_FILE=./data/CelebA_nocrop.zip
-mkdir -p ./data/
-wget -N $URL -O $ZIP_FILE
-unzip $ZIP_FILE -d ./data/
-rm $ZIP_FILE
+FILE=$1
 
-# CelebA attribute labels
-URL=https://www.dropbox.com/s/auexdy98c6g7y25/list_attr_celeba.zip?dl=0
-ZIP_FILE=./data/list_attr_celeba.zip
-wget -N $URL -O $ZIP_FILE
-unzip $ZIP_FILE -d ./data/
-rm $ZIP_FILE
+if [ $FILE == "celeba" ]; then
+
+    # CelebA images
+    URL=https://www.dropbox.com/s/3e5cmqgplchz85o/CelebA_nocrop.zip?dl=0
+    ZIP_FILE=./data/CelebA_nocrop.zip
+    mkdir -p ./data/
+    wget -N $URL -O $ZIP_FILE
+    unzip $ZIP_FILE -d ./data/
+    rm $ZIP_FILE
+
+    # CelebA attribute labels
+    URL=https://www.dropbox.com/s/auexdy98c6g7y25/list_attr_celeba.zip?dl=0
+    ZIP_FILE=./data/list_attr_celeba.zip
+    wget -N $URL -O $ZIP_FILE
+    unzip $ZIP_FILE -d ./data/
+    rm $ZIP_FILE
+
+elif [ $FILE == 'pretrained-celeba-128x128' ]; then
+
+    # StarGAN trained on CelebA (Black_Hair, Blond_Hair, Brown_Hair, Male, Young), 128x128 resolution
+    URL=https://www.dropbox.com/s/7e966qq0nlxwte4/celeba-128x128-5attrs.zip?dl=0
+    ZIP_FILE=./stargan_celeba_128/models/celeba-128x128-5attrs.zip
+    mkdir -p ./stargan_celeba_128/models/
+    wget -N $URL -O $ZIP_FILE
+    unzip $ZIP_FILE -d ./stargan_celeba_128/models/
+    rm $ZIP_FILE
+
+elif [ $FILE == 'pretrained-celeba-256x256' ]; then
+
+    # StarGAN trained on CelebA (Black_Hair, Blond_Hair, Brown_Hair, Male, Young), 256x256 resolution
+    URL=https://www.dropbox.com/s/zdq6roqf63m0v5f/celeba-256x256-5attrs.zip?dl=0
+    ZIP_FILE=./stargan_celeba_256/models/celeba-256x256-5attrs.zip
+    mkdir -p ./stargan_celeba_256/models/
+    wget -N $URL -O $ZIP_FILE
+    unzip $ZIP_FILE -d ./stargan_celeba_256/models/
+    rm $ZIP_FILE
+
+else
+    echo "Available arguments are celeba, pretrained-celeba-128x128, pretrained-celeba-256x256."
+    exit 1
+fi