24 lines
1.4 KiB
Bash
24 lines
1.4 KiB
Bash
|
set -ex
|
||
|
|
|
||
|
|
FILE=$1
|
||
|
|
|
||
|
|
if [[ $FILE != "ae_photos" && $FILE != "apple2orange" && $FILE != "summer2winter_yosemite" && $FILE != "horse2zebra" && $FILE != "monet2photo" && $FILE != "cezanne2photo" && $FILE != "ukiyoe2photo" && $FILE != "vangogh2photo" && $FILE != "maps" && $FILE != "cityscapes" && $FILE != "facades" && $FILE != "iphone2dslr_flower" && $FILE != "mini" && $FILE != "mini_pix2pix" && $FILE != "mini_colorization" && $FILE != "grumpifycat" ]]; then
|
||
|
|
echo "Available datasets are: apple2orange, summer2winter_yosemite, horse2zebra, monet2photo, cezanne2photo, ukiyoe2photo, vangogh2photo, maps, cityscapes, facades, iphone2dslr_flower, ae_photos, grumpifycat"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ $FILE == "cityscapes" ]]; then
|
||
|
|
echo "Due to license issue, we cannot provide the Cityscapes dataset from our repository. Please download the Cityscapes dataset from https://cityscapes-dataset.com, and use the script ./datasets/prepare_cityscapes_dataset.py."
|
||
|
|
echo "You need to download gtFine_trainvaltest.zip and leftImg8bit_trainvaltest.zip. For further instruction, please read ./datasets/prepare_cityscapes_dataset.py"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "Specified [$FILE]"
|
||
|
|
URL=https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/$FILE.zip
|
||
|
|
ZIP_FILE=./datasets/$FILE.zip
|
||
|
|
TARGET_DIR=./datasets/$FILE/
|
||
|
|
wget --no-check-certificate -N $URL -O $ZIP_FILE
|
||
|
|
mkdir $TARGET_DIR
|
||
|
|
unzip $ZIP_FILE -d ./datasets/
|
||
|
|
rm $ZIP_FILE
|