fixed the download link

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
 

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')