Make our datasets available for download

.gitignore

@@ -2,7 +2,6 @@ datasets/
 checkpoints/
 results/
 *.png
-!img/*.png
 */**/__pycache__
 */*.pyc
 */**/*.pyc

README.md

@@ -3,7 +3,6 @@
 
 This is our ongoing PyTorch implementation for ComboGAN.
 Code was written by [Asha Anoosheh](https://github.com/aanoosheh) (built upon [CycleGAN](https://github.com/junyanz/CycleGAN))
-(Datasets will be available soon!)
 
 
 #### [[ComboGAN Paper]](https://arxiv.org/pdf/1712.06909.pdf)
@@ -48,7 +47,9 @@ cd ComboGAN
 ```
 
 ### ComboGAN training
-Example scripts can be found in the `scripts` directory.
+Our ready datasets can be downloaded using `./datasets/download_dataset.sh <dataset_name>`.
+
+Example running scripts can be found in the `scripts` directory.
 
 - Train a model:
 ```

datasets/download_dataset.sh

@@ -0,0 +1,12 @@
+FILE=${1}
+
+if [[ $FILE != "alps_seasons" && $FILE != "painters_14" ]]; then
+    echo "Currently available datasets are: alps_seasons and painters_14"
+    exit 1
+fi
+
+URL=https://data.vision.ee.ethz.ch/cvl/combogan/$FILE.zip
+ZIP_FILE=./datasets/$FILE.zip
+wget -N $URL -O $ZIP_FILE
+unzip $ZIP_FILE -d ./datasets/
+rm $ZIP_FILE