GIST=$1

DIRNAME=${2:-./models}

if [ -z $GIST ]; then

echo "usage: download_model_from_gist.sh <gist_id> <dirname>"

exit

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

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"