From 6463af5f973990dcd9ddd3481ddd6019c4c704a3 Mon Sep 17 00:00:00 2001
From: Yunjey Choi <yunjey.choi@gmail.com>
Date: Thu, 1 Jul 2021 16:08:04 +0900
Subject: [PATCH] Add AFHQ v2 download link

---
 download.sh | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/download.sh b/download.sh
index 4588a7bb..e6006314 100644
--- a/download.sh
+++ b/download.sh
@@ -47,8 +47,16 @@ elif  [ $FILE == "afhq-dataset" ]; then
     unzip $ZIP_FILE -d ./data
     rm $ZIP_FILE
 
+elif  [ $FILE == "afhq-v2-dataset" ]; then
+    URL=https://www.dropbox.com/s/scckftx13grwmiv/afhq_v2.zip?dl=0
+    ZIP_FILE=./data/afhq_v2.zip
+    mkdir -p ./data
+    wget -N $URL -O $ZIP_FILE
+    unzip $ZIP_FILE -d ./data
+    rm $ZIP_FILE
+
 else
     echo "Available arguments are pretrained-network-celeba-hq, pretrained-network-afhq, celeba-hq-dataset, and afhq-dataset."
     exit 1
 
-fi
\ No newline at end of file
+fi