Merge pull request #1110 from sergeyk/dev

shelhamer · shelhamer · commit 58dce0ece5d9 · 2014-09-18T16:27:52.000-07:00
[model zoo] download gist script
diff --git a/docs/model_zoo.md b/docs/model_zoo.md
@@ -43,7 +43,7 @@ Github Gist is a good format for model info distribution because it can contain
 
 Try doing `scripts/upload_model_to_gist.sh models/bvlc_alexnet` to test the uploading (don't forget to delete the uploaded gist afterward).
 
-Downloading models is not yet supported as a script (there is no good commandline tool for this right now), so simply go to the Gist URL and click "Download Gist" for now.
+Downloading model info is done just as easily with `scripts/download_model_from_gist.sh <gist_id> <dirname>`.
 
 ### Hosting trained models
 
diff --git a/scripts/download_model_from_gist.sh b/scripts/download_model_from_gist.sh
@@ -0,0 +1,16 @@
+echo "usage: download_model_from_gist.sh <gist_id> <dirname>"
+
+GIST=$1
+DIRNAME=$2
+
+if [ -d "$DIRNAME/$GIST" ]; then
+    echo "$DIRNAME/$GIST already exists! Please make sure you're not overwriting anything important!"
+    exit
+fi
+
+echo "Downloading Caffe model info to $DIRNAME/$GIST ..."
+mkdir -p $DIRNAME/$GIST
+wget https://gist.github.com/$GIST/download -O $DIRNAME/$GIST/gist.tar.gz
+tar xzf $DIRNAME/$GIST/gist.tar.gz --directory=$DIRNAME/$GIST --strip-components=1
+rm $DIRNAME/$GIST/gist.tar.gz
+echo "Done"
