Merge pull request tensorflow#482 from cshallue/unzip

Use tar on OSX to unzip the MSCOCO data file.
malobato · Sep 30, 2016 · 3bdc325 · 3bdc325
2 parents 7e9b983 + 608e790
commit 3bdc325
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/im2txt/im2txt/data/download_and_preprocess_mscoco.sh b/im2txt/im2txt/data/download_and_preprocess_mscoco.sh
@@ -29,6 +29,12 @@ if [ -z "$1" ]; then
   exit
 fi
 
+if [ "$(uname)" == "Darwin" ]; then
+  UNZIP="tar -xf"
+else
+  UNZIP="unzip -nq"
+fi
+
 # Create the output directories.
 OUTPUT_DIR="${1%/}"
 SCRATCH_DIR="${OUTPUT_DIR}/raw-data"
@@ -49,7 +55,7 @@ function download_and_unzip() {
     echo "Skipping download of ${FILENAME}"
   fi
   echo "Unzipping ${FILENAME}"
-  unzip -nq ${FILENAME}
+  ${UNZIP} ${FILENAME}
 }
 
 cd ${SCRATCH_DIR}