download sample set

Ranqing · Nov 2, 2017 · 0b2e291 · 0b2e291
1 parent d7e3651
commit 0b2e291
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 2 deletions.
diff --git a/demo_dlsm.ipynb b/demo_dlsm.ipynb
@@ -37,7 +37,7 @@
    "source": [
     "# Get sample data\n",
     "!sh download_sample.sh\n",
-    "SAMPLE_DIR = 'shapenet_sample'"
+    "SAMPLE_DIR = os.path.join('data', 'shapenet_sample')"
    ]
   },
   {

diff --git a/demo_vlsm.ipynb b/demo_vlsm.ipynb
@@ -37,7 +37,7 @@
    "source": [
     "# Get sample data\n",
     "!sh download_sample.sh\n",
-    "SAMPLE_DIR = 'shapenet_sample'"
+    "SAMPLE_DIR = os.path.join('data', 'shapenet_sample')"
    ]
   },
   {

diff --git a/download_sample.sh b/download_sample.sh
@@ -0,0 +1,27 @@
+#!/bin/bash
+cd data
+URL=http://people.eecs.berkeley.edu/~akar/lsm/shapenet_sample.tar.gz
+CHECKSUM=696f75abebf387af55e8e06661162926
+FILE=shapenet_sample.tar.gz
+
+if [ -f $FILE ]; then
+  echo "File already exists. Checking md5..."
+  os=`uname -s`
+  if [ "$os" = "Linux" ]; then
+    checksum=`md5sum $FILE | awk '{ print $1 }'`
+  elif [ "$os" = "Darwin" ]; then
+    checksum=`cat $FILE | md5`
+  fi
+  if [ "$checksum" = "$CHECKSUM" ]; then
+    echo "Checksum is correct. No need to download."
+    exit 0
+  else
+    echo "Checksum is incorrect. Need to download again."
+  fi
+fi
+
+wget $URL -O $FILE
+tar xvzf $FILE
+
+cd ..
+echo "Done. Please run this command again to verify that checksum = $CHECKSUM."