| #!/usr/bin/env sh |
|
|
| GIST=$1 |
| DIRNAME=${2:-./models} |
|
|
| if [ -z $GIST ]; then |
| echo "usage: download_model_from_gist.sh <gist_id> <dirname>" |
| exit |
| fi |
|
|
| GIST_DIR=$(echo $GIST | tr '/' '-') |
| MODEL_DIR="$DIRNAME/$GIST_DIR" |
|
|
| if [ -d $MODEL_DIR ]; then |
| echo "$MODEL_DIR already exists! Please make sure you're not overwriting anything important!" |
| exit |
| fi |
|
|
| echo "Downloading Caffe model info to $MODEL_DIR ..." |
| mkdir -p $MODEL_DIR |
| wget https://gist.github.com/$GIST/download -O $MODEL_DIR/gist.zip |
| unzip -j $MODEL_DIR/gist.zip -d $MODEL_DIR |
| rm $MODEL_DIR/gist.zip |
| echo "Done" |
|
|
