updates

.gitignore

@@ -30,6 +30,7 @@ data/*
 !data/trainvalno5k.shapes
 !data/5k.shapes
 !data/5k.txt
+!data/*.sh
 
 pycocotools/*
 results*.txt

data/get_coco_dataset_gdrive.sh

@@ -0,0 +1,11 @@
+#!/bin/bash
+# https://stackoverflow.com/questions/48133080/how-to-download-a-google-drive-url-via-curl-or-wget/48133859
+
+# Download COCO dataset
+fileid="1HaXkef9z6y5l4vUnCYgdmEAj61c6bfWO"
+filename="coco_gdrive.zip"
+curl -c ./cookie -s -L "https://drive.google.com/uc?export=download&id=${fileid}" > /dev/null
+curl -Lb ./cookie "https://drive.google.com/uc?export=download&confirm=`awk '/download/ {print $NF}' ./cookie`&id=${fileid}" -o ${filename}
+
+# Unzip
+unzip -q coco_gdrive.zip