From f4a1c5873f8492b37c09b127f975172c24f2111a Mon Sep 17 00:00:00 2001
From: yunjey <yunjey47@naver.com>
Date: Tue, 22 Jan 2019 11:16:52 +0000
Subject: [PATCH] fixed the download link

---
 download.sh | 12 +++---------
 main.py     |  4 ++--
 2 files changed, 5 insertions(+), 11 deletions(-)

diff --git a/download.sh b/download.sh
index a97b7c0..7c45534 100644
--- a/download.sh
+++ b/download.sh
@@ -2,20 +2,14 @@ FILE=$1
 
 if [ $FILE == "celeba" ]; then
 
-    # CelebA images
-    URL=https://www.dropbox.com/s/3e5cmqgplchz85o/CelebA_nocrop.zip?dl=0
-    ZIP_FILE=./data/CelebA_nocrop.zip
+    # CelebA images and attribute labels
+    URL=https://www.dropbox.com/s/d1kjpkqklf0uw77/celeba.zip?dl=0
+    ZIP_FILE=./data/celeba.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
 
diff --git a/main.py b/main.py
index 5079b63..8400abb 100644
--- a/main.py
+++ b/main.py
@@ -91,8 +91,8 @@ if __name__ == '__main__':
     parser.add_argument('--use_tensorboard', type=str2bool, default=True)
 
     # Directories.
-    parser.add_argument('--celeba_image_dir', type=str, default='data/CelebA_nocrop/images')
-    parser.add_argument('--attr_path', type=str, default='data/list_attr_celeba.txt')
+    parser.add_argument('--celeba_image_dir', type=str, default='data/celeba/images')
+    parser.add_argument('--attr_path', type=str, default='data/celeba/list_attr_celeba.txt')
     parser.add_argument('--rafd_image_dir', type=str, default='data/RaFD/train')
     parser.add_argument('--log_dir', type=str, default='stargan/logs')
     parser.add_argument('--model_save_dir', type=str, default='stargan/models')