diff options
| -rw-r--r-- | docs/model_zoo.md | 2 | ||||
| -rwxr-xr-x | scripts/download_model_from_gist.sh | 16 |
2 files changed, 17 insertions, 1 deletions
diff --git a/docs/model_zoo.md b/docs/model_zoo.md index 5603c62c..330850cb 100644 --- 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 new file mode 100755 index 00000000..61d995fb --- /dev/null +++ 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" |